Spaces:

multimodalart
/

LoraTheExplorer

Running on A100

multimodalart HF staff commited on Aug 16, 2023

Commit

8df8f4d

•

1 Parent(s): 41a4356

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -83,8 +83,8 @@ def run_lora(prompt, negative, weight, selected_state):
                     pipe.unet,
                     for_inference=True,
                 )
-                lora_model = lora_model.to("cuda")
-                lora_model.apply_to(pipe.text_encoder, pipe.unet)
                 lora_model = lora_model.to("cuda")
                 lora_model.merge_to(
                     pipe.text_encoder, pipe.unet, weights_sd, torch.float16, "cuda"

                     pipe.unet,
                     for_inference=True,
                 )
+                #lora_model = lora_model.to("cuda")
+                #lora_model.apply_to(pipe.text_encoder, pipe.unet)
                 lora_model = lora_model.to("cuda")
                 lora_model.merge_to(
                     pipe.text_encoder, pipe.unet, weights_sd, torch.float16, "cuda"