diff --git a/server/text_generation_server/utils/hub.py b/server/text_generation_server/utils/hub.py index 965cae99..9443d21b 100644 --- a/server/text_generation_server/utils/hub.py +++ b/server/text_generation_server/utils/hub.py @@ -11,6 +11,7 @@ from huggingface_hub.constants import HUGGINGFACE_HUB_CACHE from huggingface_hub.utils import ( LocalEntryNotFoundError, EntryNotFoundError, # Import here to ease try/except in other part of the lib + RevisionNotFoundError ) WEIGHTS_CACHE_OVERRIDE = os.getenv("WEIGHTS_CACHE_OVERRIDE", None)