This commit is contained in:
dstnluong-google 2024-02-22 22:00:15 +00:00
parent 03fb94b853
commit 39fae920d8

View File

@ -124,7 +124,7 @@ def download_weights(
from text_generation_server import utils from text_generation_server import utils
if model_id.startswith(utils.GCS_PREFIX): if model_id.startswith(utils.GCS_PREFIX):
utisls.download_gcs_dir_to_local(model_id, utils.GCS_LOCAL_DIR) utils.download_gcs_dir_to_local(model_id, utils.GCS_LOCAL_DIR)
model_id = utils.GCS_LOCAL_DIR model_id = utils.GCS_LOCAL_DIR
# Test if files were already download # Test if files were already download