Spaces:
Runtime error
Runtime error
Update src/bart_with_group_beam.py
Browse files
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("
|
369 |
-
>>> model = AutoModelForSeq2SeqLM.from_pretrained("
|
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
|