awinml commited on
Commit
cb9b6c0
1 Parent(s): 020de1d

Upload utils.py

Browse files
Files changed (1) hide show
  1. utils.py +2 -2
utils.py CHANGED
@@ -36,8 +36,8 @@ def get_spacy_model():
36
 
37
  @st.experimental_singleton
38
  def get_flan_alpaca_xl_model():
39
- model = AutoModelForSeq2SeqLM.from_pretrained("/models/flan-alpaca-xl")
40
- tokenizer = AutoTokenizer.from_pretrained("/models/flan-alpaca-xl")
41
  return model, tokenizer
42
 
43
 
 
36
 
37
  @st.experimental_singleton
38
  def get_flan_alpaca_xl_model():
39
+ model = AutoModelForSeq2SeqLM.from_pretrained("/models/flan-alpaca-xl/")
40
+ tokenizer = AutoTokenizer.from_pretrained("/models/flan-alpaca-xl/")
41
  return model, tokenizer
42
 
43