Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
6d6bac1
1
Parent(s):
40bfaac
app.py
Browse files- app.py +1 -1
- utils/utils.py +2 -2
app.py
CHANGED
|
@@ -33,7 +33,7 @@ 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="
|
| 37 |
pipe = pipe.to(device)
|
| 38 |
return pipe
|
| 39 |
|
|
|
|
| 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="style-lineart_02.safetensors")
|
| 37 |
pipe = pipe.to(device)
|
| 38 |
return pipe
|
| 39 |
|
utils/utils.py
CHANGED
|
@@ -62,8 +62,8 @@ def load_tagger_model(model_dir):
|
|
| 62 |
|
| 63 |
def load_lora_model(model_dir):
|
| 64 |
folder = model_dir
|
| 65 |
-
file_name = '
|
| 66 |
-
url = "https://huggingface.co/
|
| 67 |
|
| 68 |
file_path = os.path.join(folder, file_name)
|
| 69 |
if not os.path.exists(file_path):
|
|
|
|
| 62 |
|
| 63 |
def load_lora_model(model_dir):
|
| 64 |
folder = model_dir
|
| 65 |
+
file_name = 'style-lineart_02.safetensors'
|
| 66 |
+
url = "https://huggingface.co/2vXpSwA7/iroiro-lora/resolve/main/test3/style-lineart_02.safetensors"
|
| 67 |
|
| 68 |
file_path = os.path.join(folder, file_name)
|
| 69 |
if not os.path.exists(file_path):
|