Update app.py

app.py

@@ -82,6 +82,7 @@ def run_lora(prompt, negative, weight, selected_state):
                     pipe.unet,
                     for_inference=True,
                 )
+                lora_model.to(device)
                 lora_model.apply_to(pipe.text_encoder, pipe.unet) #is apply too all you need?
                 #lora_model.merge_to(
                 #    pipe.text_encoder, pipe.unet, weights_sd, torch.float16, "cuda"