Fix

app_inference.py

@@ -61,6 +61,13 @@ class InferenceUtil:
         instance_prompt = getattr(card.data, 'instance_prompt', '')
         return base_model, instance_prompt
 
+    def reload_lora_model_list_and_update_model_info(
+            self, model_source: str) -> tuple[dict, str, str]:
+        model_list_update = self.reload_lora_model_list(model_source)
+        model_list = model_list_update['choices']
+        model_info = self.load_model_info(model_list[0] if model_list else '')
+        return model_list_update, *model_info
+
 
 def create_inference_demo(pipe: InferencePipeline,
                           hf_token: str | None = None) -> gr.Blocks:
@@ -117,12 +124,22 @@ def create_inference_demo(pipe: InferencePipeline,
             with gr.Column():
                 result = gr.Image(label='Result')
 
-        model_source.change(fn=app.reload_lora_model_list,
-                            inputs=model_source,
-                            outputs=lora_model_id)
-        reload_button.click(fn=app.reload_lora_model_list,
-                            inputs=model_source,
-                            outputs=lora_model_id)
+        model_source.change(
+            fn=app.reload_lora_model_list_and_update_model_info,
+            inputs=model_source,
+            outputs=[
+                lora_model_id,
+                base_model_used_for_training,
+                instance_prompt_used_for_training,
+            ])
+        reload_button.click(
+            fn=app.reload_lora_model_list_and_update_model_info,
+            inputs=model_source,
+            outputs=[
+                lora_model_id,
+                base_model_used_for_training,
+                instance_prompt_used_for_training,
+            ])
         lora_model_id.change(fn=app.load_model_info,
                              inputs=lora_model_id,
                              outputs=[