multimodalart HF staff commited on
Commit
bf42f55
1 Parent(s): a3f28a3

Update app.py

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