lhzstar commited on
Commit
31bd8d7
1 Parent(s): 2fc6e46

new commits

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -17,7 +17,7 @@ def hide_footer():
17
  st.markdown(hide_st_style, unsafe_allow_html=True)
18
 
19
  @st.cache_resource
20
- def get_seq2seq_model(model_id, use_unlimiformer=False, _tokenizer=None):
21
  model = AutoModelForSeq2SeqLM.from_pretrained(model_id)
22
  if use_unlimiformer:
23
  defaults = UnlimiformerArguments()
 
17
  st.markdown(hide_st_style, unsafe_allow_html=True)
18
 
19
  @st.cache_resource
20
+ def get_seq2seq_model(model_id, use_unlimiformer=True, _tokenizer=None):
21
  model = AutoModelForSeq2SeqLM.from_pretrained(model_id)
22
  if use_unlimiformer:
23
  defaults = UnlimiformerArguments()