multimodalart HF staff commited on
Commit
d21a36f
1 Parent(s): e5502cf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -74,7 +74,7 @@ def run_lora(prompt, negative, weight, selected_state):
74
  else:
75
  multiplier = 1.0
76
 
77
- multiplier = torch.tensor([multiplier], dtype=torch.float16, device=device)
78
  lora_model, weights_sd = lora.create_network_from_weights(
79
  multiplier,
80
  full_path_lora,
@@ -84,7 +84,7 @@ def run_lora(prompt, negative, weight, selected_state):
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
90
 
 
74
  else:
75
  multiplier = 1.0
76
 
77
+ #multiplier = torch.tensor([multiplier], dtype=torch.float16, device=device)
78
  lora_model, weights_sd = lora.create_network_from_weights(
79
  multiplier,
80
  full_path_lora,
 
84
  for_inference=True,
85
  )
86
  lora_model.apply_to(pipe.text_encoder, pipe.unet)
87
+ lora_model = lora_model.to("cuda")
88
 
89
  last_merged = True
90