Use Xet high performance mode for Transformers v5 (#35098)

Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
This commit is contained in:
Harry Mellor
2026-02-23 16:19:30 +00:00
committed by GitHub
parent a7f341c323
commit c4f38696f7

View File

@@ -80,7 +80,18 @@ def enable_hf_transfer():
pass
enable_hf_transfer()
def enable_xet_high_performance():
"""automatically activates xet high performance mode"""
if "HF_XET_HIGH_PERFORMANCE" not in os.environ:
huggingface_hub.constants.HF_XET_HIGH_PERFORMANCE = True
if hasattr(huggingface_hub.constants, "HF_XET_HIGH_PERFORMANCE"):
# Transformers v5
enable_xet_high_performance()
else:
# Transformers v4
enable_hf_transfer()
class DisabledTqdm(tqdm):