LPX55 commited on
Commit
f3de34a
·
verified ·
1 Parent(s): 8788948

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -59,8 +59,12 @@ pipe.scheduler = TCDScheduler.from_config(pipe.scheduler.config)
59
 
60
  pipe.to("cuda")
61
 
62
- pipe_inpaint = StableDiffusionXLFillPipeline.from_pretrained(
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,