awinml commited on
Commit
94ee7f8
1 Parent(s): 7527a12

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +2 -2
utils.py CHANGED
@@ -44,8 +44,8 @@ def get_t5_model():
44
  def get_flan_t5_model():
45
  return pipeline(
46
  "summarization",
47
- model="google/flan-t5-xl",
48
- tokenizer="google/flan-t5-xl",
49
  max_length=512,
50
  # length_penalty = 0
51
  )
 
44
  def get_flan_t5_model():
45
  return pipeline(
46
  "summarization",
47
+ model="google/flan-t5-large",
48
+ tokenizer="google/flan-t5-large",
49
  max_length=512,
50
  # length_penalty = 0
51
  )