app.py

app.py

@@ -33,7 +33,9 @@ def load_model(lora_dir, cn_dir):
     pipe = StableDiffusionXLControlNetImg2ImgPipeline.from_pretrained(
         "cagliostrolab/animagine-xl-3.1", controlnet=controlnet, vae=vae, torch_dtype=torch.float16
     )
-    pipe.load_lora_weights(lora_dir, weight_name="style-lineart_02.safetensors")
+    pipe.load_lora_weights(lora_dir, weight_name="sdxl_BWLine.safetensors", adapter_name="sdxl_BWLine")
+    pipe.set_adapters(["sdxl_BWLine"], adapter_weights=[1.4])
+    pipe.fuse_lora()
     pipe = pipe.to(device)
     return pipe
 

utils/utils.py

@@ -62,8 +62,8 @@ def load_tagger_model(model_dir):
 
 def load_lora_model(model_dir):
   folder = model_dir
-  file_name = 'style-lineart_02.safetensors'
-  url = "https://huggingface.co/2vXpSwA7/iroiro-lora/resolve/main/test3/style-lineart_02.safetensors"
+  file_name = 'sdxl_BWLine.safetensors'
+  url = "https://huggingface.co/tori29umai/lineart/resolve/main/sdxl_BWLine.safetensors"
 
   file_path = os.path.join(folder, file_name)
   if not os.path.exists(file_path):