From 5c2a0e45551d61679377e009003f8324c2404911 Mon Sep 17 00:00:00 2001 From: Ubuntu Date: Wed, 24 May 2023 09:46:46 +0000 Subject: [PATCH] Missing import. --- server/text_generation_server/utils/hub.py | 1 + 1 file changed, 1 insertion(+) 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)