diff --git a/server/text_generation_server/utils/gcs.py b/server/text_generation_server/utils/gcs.py index 0f55244c..993f7c39 100644 --- a/server/text_generation_server/utils/gcs.py +++ b/server/text_generation_server/utils/gcs.py @@ -1,5 +1,6 @@ import os from google.cloud import storage +from loguru import logger GCS_PREFIX = "gs://" GCS_LOCAL_DIR = "/tmp/gcs_model"