Skip to content

Commit fed0d4f

Browse files
authored
Update model.py
1 parent 6fe2187 commit fed0d4f

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

ads/aqua/model/model.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -173,10 +173,12 @@ def create(
173173
"""
174174
fine_tune_weights = []
175175
model_name = ""
176+
is_stacked = False
176177
if isinstance(model, AquaMultiModelRef):
177-
fine_tune_weights = model.fine_tune_weights
178+
fine_tune_weights = model.fine_tune_weights or []
178179
model_name = model.model_name
179180
model = model.model_id
181+
is_stacked = True
180182

181183
service_model = DataScienceModel.from_id(model)
182184
target_project = project_id or PROJECT_OCID
@@ -193,7 +195,7 @@ def create(
193195
}
194196

195197
custom_model = None
196-
if fine_tune_weights:
198+
if is_stacked:
197199
custom_model = self._create_model_group(
198200
model_id=model,
199201
model_name=model_name,
@@ -290,7 +292,7 @@ def _create_model_group(
290292
# must also include base model info in member models to create stacked model group
291293
member_models.append(
292294
{
293-
"inference_key": model_name or service_model.display_name,
295+
"inference_key": model_name,
294296
"model_id": model_id,
295297
}
296298
)

0 commit comments

Comments
 (0)