Spaces:

thekarthikeyansekar
/

sd_2_inpainting

Running

thekarthikeyansekar commited on Apr 23

Commit

15c7675

•

1 Parent(s): 55afb05

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ from diffusers import AutoPipelineForInpainting
 from diffusers.utils import load_image
 pipe = AutoPipelineForInpainting.from_pretrained("diffusers/stable-diffusion-xl-1.0-inpainting-0.1", torch_dtype=torch.float16, variant="fp16")
-generator = torch.Generator(device="cuda").manual_seed(0)
 def main():
     st.title("Image and Text Preview App")

 from diffusers.utils import load_image
 pipe = AutoPipelineForInpainting.from_pretrained("diffusers/stable-diffusion-xl-1.0-inpainting-0.1", torch_dtype=torch.float16, variant="fp16")
+generator = torch.Generator().manual_seed(0)
 def main():
     st.title("Image and Text Preview App")