Hugging Face's logo

Spaces:
multimodalart
/
face-to-all
Running on Zero

App Files Community
35
multimodalart HF Staff commited on
Commit
8df8f4d
·
1 Parent(s): 41a4356

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -83,8 +83,8 @@ def run_lora(prompt, negative, weight, selected_state):
83
  pipe.unet,
84
  for_inference=True,
85
  )
86
- lora_model = lora_model.to("cuda")
87
- lora_model.apply_to(pipe.text_encoder, pipe.unet)
88
  lora_model = lora_model.to("cuda")
89
  lora_model.merge_to(
90
  pipe.text_encoder, pipe.unet, weights_sd, torch.float16, "cuda"
 
83
  pipe.unet,
84
  for_inference=True,
85
  )
86
+ #lora_model = lora_model.to("cuda")
87
+ #lora_model.apply_to(pipe.text_encoder, pipe.unet)
88
  lora_model = lora_model.to("cuda")
89
  lora_model.merge_to(
90
  pipe.text_encoder, pipe.unet, weights_sd, torch.float16, "cuda"