mkshing commited on
Commit
b1a67fc
·
verified ·
1 Parent(s): 7277a01

Update evo_nishikie_v1.py

Browse files
Files changed (1) hide show
  1. evo_nishikie_v1.py +1 -1
evo_nishikie_v1.py CHANGED
@@ -184,10 +184,10 @@ def load_evo_nishikie(device="cuda") -> StableDiffusionXLControlNetPipeline:
184
  torch_dtype=torch.float16,
185
  variant="fp16",
186
  )
187
- pipe = pipe.to(device, dtype=torch.float16)
188
 
189
  # Load Evo-Ukiyoe weights
190
  pipe.load_lora_weights(UKIYOE_REPO)
191
  pipe.fuse_lora(lora_scale=1.0)
192
 
 
193
  return pipe
 
184
  torch_dtype=torch.float16,
185
  variant="fp16",
186
  )
 
187
 
188
  # Load Evo-Ukiyoe weights
189
  pipe.load_lora_weights(UKIYOE_REPO)
190
  pipe.fuse_lora(lora_scale=1.0)
191
 
192
+ pipe = pipe.to(device, dtype=torch.float16)
193
  return pipe