Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -59,8 +59,12 @@ pipe.scheduler = TCDScheduler.from_config(pipe.scheduler.config)
|
|
59 |
|
60 |
pipe.to("cuda")
|
61 |
|
62 |
-
|
63 |
"andro-flock/LUSTIFY-SDXL-NSFW-checkpoint-v2-0-INPAINTING",
|
|
|
|
|
|
|
|
|
64 |
torch_dtype=torch.float16,
|
65 |
vae=vae,
|
66 |
controlnet=model,
|
|
|
59 |
|
60 |
pipe.to("cuda")
|
61 |
|
62 |
+
inpaint_model = hf_hub_download(
|
63 |
"andro-flock/LUSTIFY-SDXL-NSFW-checkpoint-v2-0-INPAINTING",
|
64 |
+
filename="lustifySDXLNSFW_v20-inpainting.safetensors",
|
65 |
+
)
|
66 |
+
pipe_inpaint = StableDiffusionXLFillPipeline.from_pretrained(
|
67 |
+
inpaint_model,
|
68 |
torch_dtype=torch.float16,
|
69 |
vae=vae,
|
70 |
controlnet=model,
|