Spaces:
Running
on
Zero
Running
on
Zero
tori29umai
commited on
Commit
•
5d415f4
1
Parent(s):
d5b61f5
app.py
Browse files
app.py
CHANGED
@@ -33,9 +33,9 @@ def load_model(lora_dir, cn_dir):
|
|
33 |
pipe = StableDiffusionXLControlNetImg2ImgPipeline.from_pretrained(
|
34 |
"cagliostrolab/animagine-xl-3.1", controlnet=controlnet, vae=vae, torch_dtype=torch.float16
|
35 |
)
|
36 |
-
pipe.load_lora_weights(lora_dir, weight_name="sdxl_BW_bold_Line.safetensors"
|
37 |
-
pipe.set_adapters(["sdxl_BW_bold_Line"], adapter_weights=[1.2])
|
38 |
-
pipe.fuse_lora()
|
39 |
pipe = pipe.to(device)
|
40 |
return pipe
|
41 |
|
|
|
33 |
pipe = StableDiffusionXLControlNetImg2ImgPipeline.from_pretrained(
|
34 |
"cagliostrolab/animagine-xl-3.1", controlnet=controlnet, vae=vae, torch_dtype=torch.float16
|
35 |
)
|
36 |
+
pipe.load_lora_weights(lora_dir, weight_name="sdxl_BW_bold_Line.safetensors")
|
37 |
+
# pipe.set_adapters(["sdxl_BW_bold_Line"], adapter_weights=[1.2])
|
38 |
+
# pipe.fuse_lora()
|
39 |
pipe = pipe.to(device)
|
40 |
return pipe
|
41 |
|