Spaces:

BertChristiaens
/

ViewDiffusion

Runtime error

BertChristiaens commited on May 14, 2023

Commit

e7b2bab

1 Parent(s): 52722af

Update inference.py

Files changed (1) hide show

inference.py CHANGED Viewed

@@ -7,7 +7,7 @@ from diffusers import StableDiffusionInpaintPipeline
 from diffusers import DPMSolverMultistepScheduler, UniPCMultistepScheduler
 @torch.inference_mode()
-@st.experimental_singleton(max_entries=5)
 def get_pipeline():
     pipe = StableDiffusionInpaintPipeline.from_pretrained("stabilityai/stable-diffusion-2-inpainting",
                                                                       torch_dtype=torch.float16)

 from diffusers import DPMSolverMultistepScheduler, UniPCMultistepScheduler
 @torch.inference_mode()
+@st.cache_resource
 def get_pipeline():
     pipe = StableDiffusionInpaintPipeline.from_pretrained("stabilityai/stable-diffusion-2-inpainting",
                                                                       torch_dtype=torch.float16)