Update app.py

app.py

@@ -1,8 +1,10 @@
 
+
 import os
 import json
 import torch
 import random
+import copy
 
 import gradio as gr
 from glob import glob
@@ -147,8 +149,8 @@ class AnimateController:
             raise gr.Error(f"Please select a pretrained model path.")
         if motion_module_dropdown == "": 
             raise gr.Error(f"Please select a motion module.")
-        if base_model_dropdown == "":
-            raise gr.Error(f"Please select a base DreamBooth model.")
+        # if base_model_dropdown == "":
+        #     raise gr.Error(f"Please select a base DreamBooth model.")
 
         if is_xformers_available(): self.unet.enable_xformers_memory_efficient_attention()
 
@@ -158,11 +160,13 @@ class AnimateController:
         ).to("cuda")
         
         if self.lora_model_state_dict != {}:
-            pipeline = convert_lora(pipeline, self.lora_model_state_dict, alpha=lora_alpha_slider)
+            print(f"Lora alpha: {lora_alpha_slider}")
+            pipeline = convert_lora(copy.deepcopy(pipeline), self.lora_model_state_dict, alpha=lora_alpha_slider)
 
         pipeline.to("cuda")
 
-        if seed_textbox != -1 and seed_textbox != "": torch.manual_seed(int(seed_textbox))
+        seed_textbox = int(seed_textbox)
+        if seed_textbox != -1 and seed_textbox != "": torch.manual_seed(seed_textbox)
         else: torch.seed()
         seed = torch.initial_seed()
         
@@ -259,7 +263,7 @@ def ui():
                 )
                 lora_model_dropdown.change(fn=controller.update_lora_model, inputs=[lora_model_dropdown], outputs=[lora_model_dropdown])
                 
-                lora_alpha_slider = gr.Slider(label="LoRA alpha", value=0.8, minimum=0, maximum=2, interactive=True)
+                lora_alpha_slider = gr.Slider(label="LoRA alpha", value=0.7, minimum=0, maximum=2, interactive=True)
                 
                 personalized_refresh_button = gr.Button(value="\U0001F503", elem_classes="toolbutton")
                 def update_personalized_model():