File tree Expand file tree Collapse file tree 1 file changed +3
-1
lines changed Expand file tree Collapse file tree 1 file changed +3
-1
lines changed Original file line number Diff line number Diff line change @@ -618,7 +618,8 @@ ModelService::DownloadModelFromCortexsoAsync(
618
618
.branch_name = branch,
619
619
.path_to_model_yaml = rel.string (),
620
620
.model_alias = unique_model_id,
621
- .status = cortex::db::ModelStatus::Downloaded};
621
+ .status = cortex::db::ModelStatus::Downloaded,
622
+ .engine = mc.engine };
622
623
auto result = db_service_->AddModelEntry (model_entry);
623
624
624
625
if (result.has_error ()) {
@@ -627,6 +628,7 @@ ModelService::DownloadModelFromCortexsoAsync(
627
628
} else {
628
629
if (auto m = db_service_->GetModelInfo (unique_model_id); m.has_value ()) {
629
630
auto upd_m = m.value ();
631
+ upd_m.path_to_model_yaml = rel.string ();
630
632
upd_m.status = cortex::db::ModelStatus::Downloaded;
631
633
if (auto r = db_service_->UpdateModelEntry (unique_model_id, upd_m);
632
634
r.has_error ()) {
You can’t perform that action at this time.
0 commit comments