Upgrade transformers==5.3.0 (#17784)

Signed-off-by: Xinyuan Tong <xinyuantong.cs@gmail.com>
Co-authored-by: Kangyan-Zhou <zky314343421@gmail.com>
Co-authored-by: Alison Shao <alisonshao@mac.lan>
Co-authored-by: Mick <mickjagger19@icloud.com>
This commit is contained in:
Xinyuan Tong
2026-03-18 13:50:43 -07:00
committed by GitHub
co-authored by Kangyan-Zhou Alison Shao Mick
parent e5750a572c
commit d1e95af282
95 changed files with 1134 additions and 341 deletions
@@ -1,3 +1,4 @@
import gc
import multiprocessing
import os
import time
@@ -216,10 +217,20 @@ def _run_sglang_subprocess(
# 5 - release hf model
_mem_usage = get_gpu_memory_gb(rank)
print(f"GPU{rank} Memory usage after resuming Sgl weights: {_mem_usage}")
# In transformers v5, from_pretrained with device_map attaches accelerate
# dispatch hooks that hold strong refs to parameters. Remove them first.
try:
from accelerate.hooks import remove_hook_from_submodules
remove_hook_from_submodules(hf_model)
except (ImportError, Exception):
pass
del hf_model
hf_model = None
gc.collect()
torch.cuda.empty_cache()
time.sleep(3)
gc.collect()
torch.cuda.empty_cache()
_curr_usage = get_gpu_memory_gb(rank)
assert (