Update README.md
Browse files
README.md
CHANGED
@@ -45,7 +45,8 @@ preds = model.generate(**inputs,
|
|
45 |
no_repeat_ngram_size=2
|
46 |
)
|
47 |
|
48 |
-
decoded_preds = tokenizer.batch_decode(preds[:, inputs['input_ids'].shape[1]:],
|
|
|
49 |
output = prompt + ' ' + decoded_preds.strip()
|
50 |
print(output)
|
51 |
```
|
|
|
45 |
no_repeat_ngram_size=2
|
46 |
)
|
47 |
|
48 |
+
decoded_preds = tokenizer.batch_decode(preds[:, inputs['input_ids'].shape[1]:],
|
49 |
+
skip_special_tokens=True)[0]
|
50 |
output = prompt + ' ' + decoded_preds.strip()
|
51 |
print(output)
|
52 |
```
|