multimodalart HF staff commited on
Commit
41a4356
β€’
1 Parent(s): ce92feb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -86,6 +86,9 @@ def run_lora(prompt, negative, weight, selected_state):
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
  last_merged = True
90
 
91
  image = pipe(
 
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"
91
+ )
92
  last_merged = True
93
 
94
  image = pipe(