Update evo_nishikie_v1.py
Browse files- 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
|