face-to-all

Runtime error

multimodalart HF Staff commited on Aug 16, 2023

Commit

fc8ab35

1 Parent(s): b52bfac

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -85,7 +85,7 @@ def run_lora(prompt, negative, weight, selected_state):
                 )
                 lora_model = lora_model.to("cuda")
                 lora_model.apply_to(pipe.text_encoder, pipe.unet)
             last_merged = True
     image = pipe(

                 )
                 lora_model = lora_model.to("cuda")
                 lora_model.apply_to(pipe.text_encoder, pipe.unet)
+                lora_model = lora_model.to("cuda")
             last_merged = True
     image = pipe(