Garbage collector regression in the online server (#10621)
This commit is contained in:
@@ -100,6 +100,7 @@ def freeze_gc(enable_cudagraph_gc: bool):
|
|||||||
finally:
|
finally:
|
||||||
if should_freeze:
|
if should_freeze:
|
||||||
gc.unfreeze()
|
gc.unfreeze()
|
||||||
|
gc.collect()
|
||||||
|
|
||||||
|
|
||||||
def _to_torch(model: torch.nn.Module, reverse: bool, num_tokens: int):
|
def _to_torch(model: torch.nn.Module, reverse: bool, num_tokens: int):
|
||||||
|
|||||||
Reference in New Issue
Block a user