BertChristiaens commited on
Commit
ba749e7
1 Parent(s): 47d7756

Update pipelines.py

Browse files
Files changed (1) hide show
  1. pipelines.py +2 -2
pipelines.py CHANGED
@@ -105,7 +105,7 @@ class SDPipeline:
105
 
106
 
107
 
108
- @st.experimental_singleton(max_entries=5)
109
  def get_controlnet():
110
  """Method to load the controlnet model
111
  Returns:
@@ -116,7 +116,7 @@ def get_controlnet():
116
 
117
 
118
 
119
- @st.experimental_singleton(max_entries=5)
120
  def get_inpainting_pipeline():
121
  """Method to load the inpainting pipeline
122
  Returns:
 
105
 
106
 
107
 
108
+ @st.cache_resource(max_entries=5)
109
  def get_controlnet():
110
  """Method to load the controlnet model
111
  Returns:
 
116
 
117
 
118
 
119
+ @st.cache_resource(max_entries=5)
120
  def get_inpainting_pipeline():
121
  """Method to load the inpainting pipeline
122
  Returns: