File tree Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Original file line number Diff line number Diff 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 )
You can’t perform that action at this time.
0 commit comments