Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
  • Loading branch information
WrRan authored and amyeroberts committed Nov 3, 2022
1 parent b933314 commit 98d388b
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -568,7 +568,7 @@ def forward(
>>> input_ids = tokenizer("This is a really long text", 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=labels)
>>> loss, logits = outputs.loss, outputs.logits
>>> # save and load from pretrained
Expand Down

0 comments on commit 98d388b

Please sign in to comment.