Spaces:
Runtime error
Runtime error
lhzstar
commited on
Commit
•
31bd8d7
1
Parent(s):
2fc6e46
new commits
Browse files
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=
|
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()
|