BertChristiaens commited on
Commit
e7b2bab
1 Parent(s): 52722af

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -7,7 +7,7 @@ from diffusers import StableDiffusionInpaintPipeline
7
  from diffusers import DPMSolverMultistepScheduler, UniPCMultistepScheduler
8
 
9
  @torch.inference_mode()
10
- @st.experimental_singleton(max_entries=5)
11
  def get_pipeline():
12
  pipe = StableDiffusionInpaintPipeline.from_pretrained("stabilityai/stable-diffusion-2-inpainting",
13
  torch_dtype=torch.float16)
 
7
  from diffusers import DPMSolverMultistepScheduler, UniPCMultistepScheduler
8
 
9
  @torch.inference_mode()
10
+ @st.cache_resource
11
  def get_pipeline():
12
  pipe = StableDiffusionInpaintPipeline.from_pretrained("stabilityai/stable-diffusion-2-inpainting",
13
  torch_dtype=torch.float16)