Initial commitcq

app.py

@@ -72,8 +72,8 @@ def generate_image(prompt, seed, num_steps, guidance_scale, eta):
         )
         pipe.scheduler = TCDScheduler.from_config(pipe.scheduler.config)
         
-        # Load and fuse LoRA weights
-        pipe.load_lora_weights(tcd_lora_id)
+        # Load and fuse LoRA weights with prefix=None
+        pipe.load_lora_weights(tcd_lora_id, prefix=None)
         pipe.fuse_lora()
         
         # Generate the image
@@ -103,8 +103,8 @@ def generate_community_image(prompt, model_name, seed, num_steps, guidance_scale
         )
         pipe.scheduler = TCDScheduler.from_config(pipe.scheduler.config)
         
-        # Load and fuse LoRA weights
-        pipe.load_lora_weights(tcd_lora_id)
+        # Load and fuse LoRA weights with prefix=None
+        pipe.load_lora_weights(tcd_lora_id, prefix=None)
         pipe.fuse_lora()
         
         # Generate the image
@@ -135,9 +135,9 @@ def generate_style_mix(prompt, seed, num_steps, guidance_scale, eta, style_weigh
         )
         pipe.scheduler = TCDScheduler.from_config(pipe.scheduler.config)
         
-        # Load multiple LoRA weights
-        pipe.load_lora_weights(tcd_lora_id, adapter_name="tcd")
-        pipe.load_lora_weights(styled_lora_id, adapter_name="style")
+        # Load multiple LoRA weights with prefix=None
+        pipe.load_lora_weights(tcd_lora_id, adapter_name="tcd", prefix=None)
+        pipe.load_lora_weights(styled_lora_id, adapter_name="style", prefix=None)
         
         # Set adapter weights
         pipe.set_adapters(["tcd", "style"], adapter_weights=[1.0, style_weight])
@@ -177,8 +177,8 @@ def generate_controlnet(prompt, init_image, seed, num_steps, guidance_scale, eta
         )
         pipe.scheduler = TCDScheduler.from_config(pipe.scheduler.config)
         
-        # Load and fuse LoRA weights
-        pipe.load_lora_weights(tcd_lora_id)
+        # Load and fuse LoRA weights with prefix=None
+        pipe.load_lora_weights(tcd_lora_id, prefix=None)
         pipe.fuse_lora()
         
         # Generate depth map
@@ -215,8 +215,8 @@ def inpaint_image(prompt, init_image, mask_image, seed, num_steps, guidance_scal
         )
         pipe.scheduler = TCDScheduler.from_config(pipe.scheduler.config)
         
-        # Load and fuse LoRA weights
-        pipe.load_lora_weights(tcd_lora_id)
+        # Load and fuse LoRA weights with prefix=None
+        pipe.load_lora_weights(tcd_lora_id, prefix=None)
         pipe.fuse_lora()
         
         # Generate the image
@@ -261,11 +261,12 @@ def generate_animation(prompt, seed, num_steps, guidance_scale, eta, num_frames,
         # Set TCD scheduler
         pipe.scheduler = TCDScheduler.from_config(pipe.scheduler.config)
         
-        # Load LoRA weights
-        pipe.load_lora_weights(tcd_lora_id, adapter_name="tcd")
+        # Load LoRA weights with prefix=None
+        pipe.load_lora_weights(tcd_lora_id, adapter_name="tcd", prefix=None)
         pipe.load_lora_weights(
             motion_lora_id,
-            adapter_name="motion-lora"
+            adapter_name="motion-lora",
+            prefix=None
         )
         
         # Set adapter weights

requirements.txt

@@ -2,7 +2,7 @@ torch>=2.0.0
 diffusers>=0.24.0
 transformers>=4.36.0
 accelerate>=0.25.0
-gradio>=4.19.2
+gradio==4.44.1
 safetensors>=0.4.0 
 peft>=0.7.0
 requests>=2.31.0