Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
322677e
1
Parent(s):
1f4dfdd
Update app.py
Browse files
app.py
CHANGED
|
@@ -34,7 +34,7 @@ def load_model(lora_dir, cn_dir):
|
|
| 34 |
"cagliostrolab/animagine-xl-3.1", controlnet=controlnet, vae=vae, torch_dtype=torch.float16
|
| 35 |
)
|
| 36 |
pipe.enable_model_cpu_offload()
|
| 37 |
-
pipe.load_lora_weights(lora_dir, weight_name="
|
| 38 |
return pipe
|
| 39 |
|
| 40 |
@spaces.GPU(duration=120)
|
|
|
|
| 34 |
"cagliostrolab/animagine-xl-3.1", controlnet=controlnet, vae=vae, torch_dtype=torch.float16
|
| 35 |
)
|
| 36 |
pipe.enable_model_cpu_offload()
|
| 37 |
+
pipe.load_lora_weights(lora_dir, weight_name="syoujomannga_line.safetensors")
|
| 38 |
return pipe
|
| 39 |
|
| 40 |
@spaces.GPU(duration=120)
|