piazzola commited on
Commit
9c1deab
1 Parent(s): 425787e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -16,7 +16,7 @@ should probably proofread and complete it, then remove this comment. -->
16
  # Load model directly
17
 
18
  ```
19
- `from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
20
  import torch
21
 
22
  tokenizer = AutoTokenizer.from_pretrained("piazzola/test1")
@@ -29,7 +29,7 @@ with torch.no_grad():
29
  outputs = model.generate(**inputs)
30
  generated_text = tokenizer.decode(outputs[0])
31
 
32
- print(generated_text)`
33
  ```
34
 
35
  # checkpoint
 
16
  # Load model directly
17
 
18
  ```
19
+ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
20
  import torch
21
 
22
  tokenizer = AutoTokenizer.from_pretrained("piazzola/test1")
 
29
  outputs = model.generate(**inputs)
30
  generated_text = tokenizer.decode(outputs[0])
31
 
32
+ print(generated_text)
33
  ```
34
 
35
  # checkpoint