tori29umai commited on
Commit
5d415f4
1 Parent(s): d5b61f5
Files changed (1) hide show
  1. app.py +3 -3
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", adapter_name="sdxl_BW_bold_Line")
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