andreslu commited on
Commit
7f951b3
·
1 Parent(s): 2b6af87

Update src/bart_with_group_beam.py

Browse files
Files changed (1) hide show
  1. src/bart_with_group_beam.py +2 -2
src/bart_with_group_beam.py CHANGED
@@ -87,8 +87,8 @@ class BartForConditionalGeneration_GroupBeam(BartForConditionalGeneration):
87
  ... )
88
  >>> import torch
89
 
90
- >>> tokenizer = AutoTokenizer.from_pretrained("t5-base")
91
- >>> model = AutoModelForSeq2SeqLM.from_pretrained("t5-base")
92
 
93
  >>> encoder_input_str = "translate English to German: How old are you?"
94
  >>> encoder_input_ids = tokenizer(encoder_input_str, return_tensors="pt").input_ids
 
87
  ... )
88
  >>> import torch
89
 
90
+ >>> tokenizer = AutoTokenizer.from_pretrained("facebook/bart-large")
91
+ >>> model = AutoModelForSeq2SeqLM.from_pretrained("facebook/bart-large")
92
 
93
  >>> encoder_input_str = "translate English to German: How old are you?"
94
  >>> encoder_input_ids = tokenizer(encoder_input_str, return_tensors="pt").input_ids