Skip to content

Commit b3de9b8

Browse files
dangkai4udangkai.dk
authored andcommitted
fix empty_cache error in pt_weights_iterator (sgl-project#5151)
Co-authored-by: dangkai.dk <[email protected]>
1 parent 7a0e1e2 commit b3de9b8

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

python/sglang/srt/model_loader/weight_utils.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -460,7 +460,6 @@ def pt_weights_iterator(
460460
state = torch.load(bin_file, map_location="cpu", weights_only=True)
461461
yield from state.items()
462462
del state
463-
torch.cuda.empty_cache()
464463

465464

466465
def get_gguf_extra_tensor_names(

0 commit comments

Comments
 (0)