Update README.md
Browse files
README.md
CHANGED
|
@@ -24,4 +24,5 @@ prompt = <BOS> + title + \
|
|
| 24 |
generated = torch.tensor(tokenizer.encode(prompt)).unsqueeze(0)
|
| 25 |
|
| 26 |
model.eval();
|
| 27 |
-
```
|
|
|
|
|
|
| 24 |
generated = torch.tensor(tokenizer.encode(prompt)).unsqueeze(0)
|
| 25 |
|
| 26 |
model.eval();
|
| 27 |
+
```
|
| 28 |
+
Currently only works when resizing the model with model.resize_token_embeddings(len(tokenizer))
|