From 513c9b0244825b17f8d3dd2046d8f9277d6fc272 Mon Sep 17 00:00:00 2001 From: Jerry Chen Date: Thu, 18 Jan 2024 09:39:15 -0800 Subject: [PATCH] Update llmfoundry/utils/model_download_utils.py Co-authored-by: Irene Dea --- llmfoundry/utils/model_download_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llmfoundry/utils/model_download_utils.py b/llmfoundry/utils/model_download_utils.py index 17f0914ac4..5d8a413d91 100644 --- a/llmfoundry/utils/model_download_utils.py +++ b/llmfoundry/utils/model_download_utils.py @@ -253,7 +253,7 @@ def _read_secrets_file(secret_file_path: str,): with open(config_file, 'r', encoding='utf-8') as f: configs = yaml.safe_load(f.read()) - path = configs[model] + path = configs['models'][model] registry = secrets['registry'] def get_oras_cmd(username: Optional[str] = None,