Jorge12344321 commited on
Commit
6d1a3ed
·
1 Parent(s): dab8260

fix: remove device kwarg not supported by T5ForConditionalGeneration

Browse files
Files changed (1) hide show
  1. versions/hybrid_v4/app_v4.py +1 -3
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)