flux-lab-light

Sleeping

vilarin commited on Sep 3

Commit

788611e

•

1 Parent(s): 0784aaa

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -66,7 +66,7 @@ def enable_lora(lora_scale, lora_in, lora_add):
             lora_in = lora_add
         url = f'https://huggingface.co/{lora_in}/tree/main'
         lora_name = scrape_lora_link(url)
-        pipe.load_lora_weights(lora_add, weight_name=lora_name)
         pipe.fuse_lora(lora_scale=lora_scale)
 def generate_image(

             lora_in = lora_add
         url = f'https://huggingface.co/{lora_in}/tree/main'
         lora_name = scrape_lora_link(url)
+        pipe.load_lora_weights(lora_in, weight_name=lora_name)
         pipe.fuse_lora(lora_scale=lora_scale)
 def generate_image(