multimodalart HF staff commited on
Commit
0df5d5a
1 Parent(s): b888bcf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -81,10 +81,10 @@ def run_lora(prompt, negative, weight, selected_state):
81
  pipe.unet,
82
  for_inference=True,
83
  )
84
- lora_model.apply_to(pipe.text_encoder, pipe.unet)
85
- lora_model.merge_to(
86
- pipe.text_encoder, pipe.unet, weights_sd, torch.float16, "cuda"
87
- )
88
  last_merged = True
89
 
90
  image = pipe(
 
81
  pipe.unet,
82
  for_inference=True,
83
  )
84
+ lora_model.apply_to(pipe.text_encoder, pipe.unet) #is apply too all you need?
85
+ #lora_model.merge_to(
86
+ # pipe.text_encoder, pipe.unet, weights_sd, torch.float16, "cuda"
87
+ #)
88
  last_merged = True
89
 
90
  image = pipe(