dennispark commited on
Commit
bb24520
β€’
1 Parent(s): 0955864

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -25,7 +25,7 @@ model = T5ForConditionalGeneration.from_pretrained('paust/pko-t5-large')
25
 
26
  input_ids = tokenizer(["nsmc sentence: λ‹Ήμ‹ μ˜ 이름은 λ¬΄μ—‡μΈκ°€μš”?"]).input_ids
27
  labels = tokenizer(["T5 μž…λ‹ˆλ‹€."]).input_ids
28
- outputs = model(input_ids, labels)
29
 
30
  print(f"loss={outputs.loss} logits={outputs.logits}")
31
  ```
 
25
 
26
  input_ids = tokenizer(["nsmc sentence: λ‹Ήμ‹ μ˜ 이름은 λ¬΄μ—‡μΈκ°€μš”?"]).input_ids
27
  labels = tokenizer(["T5 μž…λ‹ˆλ‹€."]).input_ids
28
+ outputs = model(input_ids=input_ids, labels=labels)
29
 
30
  print(f"loss={outputs.loss} logits={outputs.logits}")
31
  ```