Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
08816de1
Unverified
Commit
08816de1
authored
Nov 18, 2021
by
NielsRogge
Committed by
GitHub
Nov 18, 2021
Browse files
Fix code example (#14441)
parent
01f8e639
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
3 deletions
+3
-3
src/transformers/models/encoder_decoder/modeling_encoder_decoder.py
...ormers/models/encoder_decoder/modeling_encoder_decoder.py
+3
-3
No files found.
src/transformers/models/encoder_decoder/modeling_encoder_decoder.py
View file @
08816de1
...
@@ -430,8 +430,8 @@ class EncoderDecoderModel(PreTrainedModel):
...
@@ -430,8 +430,8 @@ class EncoderDecoderModel(PreTrainedModel):
>>> model.config.pad_token_id = tokenizer.pad_token_id
>>> model.config.pad_token_id = tokenizer.pad_token_id
>>> model.config.vocab_size = model.config.decoder.vocab_size
>>> model.config.vocab_size = model.config.decoder.vocab_size
>>> input_ids = tokenizer("
Hello, my dog is cute
", return_tensors="pt").input_ids
>>> input_ids = tokenizer("
This is a really long text
", return_tensors="pt").input_ids
>>> labels = tokenizer("
Salut, mon chien est mignon
", return_tensors="pt").input_ids
>>> labels = tokenizer("
This is the corresponding summary
", return_tensors="pt").input_ids
>>> outputs = model(input_ids=input_ids, labels=input_ids)
>>> outputs = model(input_ids=input_ids, labels=input_ids)
>>> loss, logits = outputs.loss, outputs.logits
>>> loss, logits = outputs.loss, outputs.logits
...
@@ -440,7 +440,7 @@ class EncoderDecoderModel(PreTrainedModel):
...
@@ -440,7 +440,7 @@ class EncoderDecoderModel(PreTrainedModel):
>>> model = EncoderDecoderModel.from_pretrained("bert2bert")
>>> model = EncoderDecoderModel.from_pretrained("bert2bert")
>>> # generation
>>> # generation
>>> generated = model.generate(input_ids
, decoder_start_token_id=model.config.decoder.pad_token_id
)
>>> generated = model.generate(input_ids)
"""
"""
return_dict
=
return_dict
if
return_dict
is
not
None
else
self
.
config
.
use_return_dict
return_dict
=
return_dict
if
return_dict
is
not
None
else
self
.
config
.
use_return_dict
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment