Skip to content

GitLab

Unverified Commit a604f05c authored by Michael Feil's avatar Michael Feil Committed by GitHub
Browse files

improve hf_hub activation (#1438)

parent f3b79170
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 5 deletions
lm_eval/models/__init__.py
View file @ a604f05c
...@@ -11,12 +11,11 @@ from . import neuron_optimum ...@@ -11,12 +11,11 @@ from . import neuron_optimum
# TODO: implement __all__ # TODO: implement __all__
import os
try: try:
# enabling faster model download # enable hf hub transfer if available
import hf_transfer import hf_transfer # type: ignore # noqa
import huggingface_hub.constants # type: ignore
os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1" huggingface_hub.constants.HF_HUB_ENABLE_HF_TRANSFER = True
except ImportError: except ImportError:
pass pass
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment