multimodalart HF staff commited on
Commit
e5502cf
β€’
1 Parent(s): 722e096

Update app.py

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