andreslu commited on
Commit
8baa350
1 Parent(s): cbed089

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
@@ -365,8 +365,8 @@ class BartForConditionalGeneration_GroupBeam(BartForConditionalGeneration):
365
  ... )
366
  >>> import torch
367
 
368
- >>> tokenizer = AutoTokenizer.from_pretrained("t5-base")
369
- >>> model = AutoModelForSeq2SeqLM.from_pretrained("t5-base")
370
 
371
  >>> encoder_input_str = "translate English to German: How old are you?"
372
  >>> encoder_input_ids = tokenizer(encoder_input_str, return_tensors="pt").input_ids
 
365
  ... )
366
  >>> import torch
367
 
368
+ >>> tokenizer = AutoTokenizer.from_pretrained("facebook/bart-large")
369
+ >>> model = AutoModelForSeq2SeqLM.from_pretrained("facebook/bart-large")
370
 
371
  >>> encoder_input_str = "translate English to German: How old are you?"
372
  >>> encoder_input_ids = tokenizer(encoder_input_str, return_tensors="pt").input_ids