app.py

app.py

@@ -35,8 +35,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="sdxl-lineart_11.safetensors")
-    pipe.set_adapters(["sdxl-lineart_11"], adapter_weights=[1.2])
+    pipe.load_lora_weights(lora_dir, weight_name="sdxl_BWLine.safetensors")
+    pipe.set_adapters(["sdxl_BWLine"], adapter_weights=[1.4])
     pipe.fuse_lora()
     pipe = pipe.to(device)
     return pipe

utils/dl_utils.py

@@ -57,10 +57,10 @@ def dl_tagger_model(model_dir):
 
 
 def dl_lora_model(model_dir):
-    file_name = 'sdxl-lineart_11.safetensors'
+    file_name = 'sdxl_BWLine.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/sdxl/sdxl-lineart_11.safetensors"
+        url = "https://huggingface.co/tori29umai/lineart/resolve/main/sdxl_BWLine.safetensors"
         response = requests.get(url, allow_redirects=True)
         if response.status_code == 200:
             with open(file_path, 'wb') as f: