Spaces:
Sleeping
Sleeping
Commit ·
6d1a3ed
1
Parent(s): dab8260
fix: remove device kwarg not supported by T5ForConditionalGeneration
Browse files
versions/hybrid_v4/app_v4.py
CHANGED
|
@@ -157,9 +157,7 @@ st.markdown("""
|
|
| 157 |
def load_chronos_pipeline(model_name):
|
| 158 |
from chronos import ChronosPipeline
|
| 159 |
hf_name = MODELS[model_name]
|
| 160 |
-
return ChronosPipeline.from_pretrained(
|
| 161 |
-
hf_name, device="cpu",
|
| 162 |
-
)
|
| 163 |
|
| 164 |
|
| 165 |
@st.cache_data(ttl=600, show_spinner=False)
|
|
|
|
| 157 |
def load_chronos_pipeline(model_name):
|
| 158 |
from chronos import ChronosPipeline
|
| 159 |
hf_name = MODELS[model_name]
|
| 160 |
+
return ChronosPipeline.from_pretrained(hf_name)
|
|
|
|
|
|
|
| 161 |
|
| 162 |
|
| 163 |
@st.cache_data(ttl=600, show_spinner=False)
|