Update README.md
Browse files
README.md
CHANGED
@@ -45,5 +45,5 @@ with torch.no_grad():
|
|
45 |
repetition_penalty=1.15
|
46 |
).cuda()
|
47 |
output = tokenizer.decode(output_ids[0], skip_special_tokens=True)
|
48 |
-
print(output.replace(
|
49 |
```
|
|
|
45 |
repetition_penalty=1.15
|
46 |
).cuda()
|
47 |
output = tokenizer.decode(output_ids[0], skip_special_tokens=True)
|
48 |
+
print(output.replace(prompt, '').strip())
|
49 |
```
|