multimodalart HF staff commited on
Commit
818277f
1 Parent(s): 5b12685

Update app.py

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