Spaces:

multimodalart
/

lora-roulette

Running on Zero

multimodalart HF staff commited on Sep 20, 2023

Commit

90789f2

1 Parent(s): 3ff5e41

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -100,9 +100,9 @@ def merge_and_run(prompt, negative_prompt, shuffled_items, lora_1_scale=0.5, lor
   pipe = copy.deepcopy(original_pipe)
   pipe.to("cuda")
   print("Loading LoRA weights...")
-  pipe.load_lora_weights(state_dict_1.to("cuda"))
   pipe.fuse_lora(lora_1_scale)
-  pipe.load_lora_weights(state_dict_2.to("cuda"))
   pipe.fuse_lora(lora_2_scale)
   print("Loaded LoRA weights.")
   if negative_prompt == "":

   pipe = copy.deepcopy(original_pipe)
   pipe.to("cuda")
   print("Loading LoRA weights...")
+  pipe.load_lora_weights(state_dict_1)
   pipe.fuse_lora(lora_1_scale)
+  pipe.load_lora_weights(state_dict_2)
   pipe.fuse_lora(lora_2_scale)
   print("Loaded LoRA weights.")
   if negative_prompt == "":