app.py

app.py

@@ -33,8 +33,8 @@ 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", adapter_name="style-lineart_02")
-    pipe.set_adapters(["style-lineart_02"], adapter_weights=[1.2])
+    pipe.load_lora_weights(lora_dir, weight_name="sdxl_BW_bold_Line.safetensors", adapter_name="sdxl_BW_bold_Line")
+    pipe.set_adapters(["sdxl_BW_bold_Line"], adapter_weights=[1.2])
     pipe.fuse_lora()
     pipe = pipe.to(device)
     return pipe

utils/utils.py

@@ -57,10 +57,10 @@ def load_tagger_model(model_dir):
 
 
 def load_lora_model(model_dir):
-    file_name = 'style-lineart_02.safetensors'
+    file_name = 'sdxl_BW_bold_Line.safetensors'
     file_path = os.path.join(model_dir, file_name)
     if not os.path.exists(file_path):
-        url = "https://huggingface.co/2vXpSwA7/iroiro-lora/resolve/main/test3/style-lineart_02.safetensors"
+        url = "https://huggingface.co/tori29umai/lineart/blob/main/sdxl_BW_bold_Line.safetensors"
         response = requests.get(url, allow_redirects=True)
         if response.status_code == 200:
             with open(file_path, 'wb') as f: