Update llmfoundry/utils/model_download_utils.py

Co-authored-by: Irene Dea <[email protected]>
mosaicml · Jan 18, 2024 · 513c9b0 · 513c9b0
1 parent 697638f
commit 513c9b0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,