Skip to content

GitLab

Unverified Commit 7cb0572c authored by Parthe Pandit's avatar Parthe Pandit Committed by GitHub
Browse files

Update modeling_bert.py (#6897) 

outptus -> outputs in example of BertForPreTraining
parent e3c55ceb
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/modeling_bert.py
View file @ 7cb0572c
...@@ -909,7 +909,7 @@ class BertForPreTraining(BertPreTrainedModel): ...@@ -909,7 +909,7 @@ class BertForPreTraining(BertPreTrainedModel):
>>> inputs = tokenizer("Hello, my dog is cute", return_tensors="pt") >>> inputs = tokenizer("Hello, my dog is cute", return_tensors="pt")
>>> outputs = model(**inputs) >>> outputs = model(**inputs)
>>> prediction_logits = outptus.prediction_logits >>> prediction_logits = outputs.prediction_logits
>>> seq_relationship_logits = outputs.seq_relationship_logits >>> seq_relationship_logits = outputs.seq_relationship_logits
""" """
if "masked_lm_labels" in kwargs: if "masked_lm_labels" in kwargs:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment