Addressing review comments

oracle · Feb 5, 2025 · ef2c40e · ef2c40e
1 parent e449f0a
commit ef2c40e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/ads/aqua/app.py b/ads/aqua/app.py
@@ -313,10 +313,11 @@ def get_config(
                 f"Base model found for the model: {oci_model.id}. "
                 f"Loading {config_file_name} for base model {base_model_ocid}."
             )
-            base_model = self.ds_client.get_model(base_model_ocid).data
-            artifact_path = get_artifact_path(base_model.custom_metadata_list)
             if config_folder == ConfigFolder.ARTIFACT:
                 artifact_path = get_artifact_path(oci_model.custom_metadata_list)
+            else:
+                base_model = self.ds_client.get_model(base_model_ocid).data
+                artifact_path = get_artifact_path(base_model.custom_metadata_list)
         else:
             logger.info(f"Loading {config_file_name} for model {oci_model.id}...")
             artifact_path = get_artifact_path(oci_model.custom_metadata_list)