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