BertChristiaens
commited on
Commit
•
ba749e7
1
Parent(s):
47d7756
Update pipelines.py
Browse files- pipelines.py +2 -2
pipelines.py
CHANGED
@@ -105,7 +105,7 @@ class SDPipeline:
|
|
105 |
|
106 |
|
107 |
|
108 |
-
@st.
|
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.
|
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:
|