Update README.md
Browse files
README.md
CHANGED
@@ -16,7 +16,7 @@ should probably proofread and complete it, then remove this comment. -->
|
|
16 |
# Load model directly
|
17 |
|
18 |
```
|
19 |
-
|
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
|