Skip to content

Commit 3b49d21

Browse files
authored
Update entities.py
1 parent 7e278bf commit 3b49d21

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

ads/aqua/modeldeployment/entities.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -742,7 +742,13 @@ def validate_base_model(self, model_id: str) -> Union[str, AquaMultiModelRef]:
742742
self.models = [
743743
AquaMultiModelRef(
744744
model_id=segments[0],
745-
fine_tune_weights=[LoraModuleSpec(model_id=base_model.id)],
745+
model_name=segments[1],
746+
fine_tune_weights=[
747+
LoraModuleSpec(
748+
model_id=base_model.id,
749+
model_name=base_model.display_name,
750+
)
751+
],
746752
)
747753
]
748754
return self.models[0]

0 commit comments

Comments
 (0)