Spaces:
Running
on
A10G
Running
on
A10G
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:
|