Update README.md
Browse files
README.md
CHANGED
@@ -59,5 +59,5 @@ with torch.no_grad():
|
|
59 |
repetition_penalty=1.15
|
60 |
).cuda()
|
61 |
output = tokenizer.decode(output_ids[0], skip_special_tokens=True)
|
62 |
-
print(output.replace(
|
63 |
```
|
|
|
59 |
repetition_penalty=1.15
|
60 |
).cuda()
|
61 |
output = tokenizer.decode(output_ids[0], skip_special_tokens=True)
|
62 |
+
print(output.replace(prompt, '').strip())
|
63 |
```
|