fix displayname of the diffusion model

fszontagh · Jan 8, 2025 · 3216b14 · 3216b14
1 parent 71e43c8
commit 3216b14
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/ui/MainWindowUI.cpp b/src/ui/MainWindowUI.cpp
@@ -1066,7 +1066,7 @@ void MainWindowUI::onGenerate(wxCommandEvent& event) {
 
     if (type == QM::GenerationMode::TXT2IMG || type == QM::GenerationMode::IMG2IMG) {
         if (diffusionModel.empty() == false) {
-            item->model = this->m_filePickerDiffusionModel->GetFileName().GetName().utf8_string();
+            item->model = wxFileName(item->params.diffusion_model_path).GetFullName();
         }
 
         auto slgscale = this->slgScale->GetValue();
@@ -1498,6 +1498,7 @@ void MainWindowUI::OnModelFavoriteChange(wxCommandEvent& event) {
         this->treeListManager->ChangeText(modelInfo->path, modelInfo->name, 0);
         modelInfo->tags = modelInfo->tags & ~sd_gui_utils::ModelInfoTag::Favorite;
     }
+    this->treeListManager->SelectItemByModelPath(modelInfo->path);
     this->ModelManager->UpdateInfo(modelInfo);
     if (this->mapp->cfg->favorite_models_only) {
         // check if in it the m_model, i = 0 is placeholder