Update README.md
Browse files
README.md
CHANGED
@@ -69,6 +69,8 @@ with torch.no_grad():
|
|
69 |
top_p=0.9,
|
70 |
repetition_penalty=1.15
|
71 |
).cuda()
|
|
|
|
|
72 |
output = tokenizer.decode(output_ids[0], skip_special_tokens=True)
|
73 |
print(output.replace(text, '').strip())
|
74 |
```
|
|
|
69 |
top_p=0.9,
|
70 |
repetition_penalty=1.15
|
71 |
).cuda()
|
72 |
+
|
73 |
+
|
74 |
output = tokenizer.decode(output_ids[0], skip_special_tokens=True)
|
75 |
print(output.replace(text, '').strip())
|
76 |
```
|