Spaces:

ehristoforu
/

Visionix-Playground

Paused

ehristoforu commited on Jun 10, 2024

Commit

4ac21b5

verified ·

1 Parent(s): a1f4d16

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -36,7 +36,7 @@ pipe = StableDiffusionXLPipeline.from_pretrained(
 pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
 pipe.to('cuda')
 pipe_inpaint = StableDiffusionXLInpaintPipeline.from_pretrained(
-    "ehristoforu/Visionix-alpha-inpainting",
     torch_dtype=torch.float16,
     use_safetensors=True,
 )

 pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
 pipe.to('cuda')
 pipe_inpaint = StableDiffusionXLInpaintPipeline.from_pretrained(
+    "https://huggingface.co/ehristoforu/Visionix-alpha-inpainting/blob/main/Visionix-alpha-inpainting.safetensors",
     torch_dtype=torch.float16,
     use_safetensors=True,
 )