Update README.md
Browse files
README.md
CHANGED
|
@@ -25,5 +25,6 @@ x = tokenizer(prompt, max_length=1024, truncation=True, padding=True, return_ten
|
|
| 25 |
y = model.generate(x, max_length=1024)[0]
|
| 26 |
y_dec = tokenizer.decode(y, max_length=1024, truncation=True)
|
| 27 |
output = y_dec.split('###[Output]:\n')[1].split('<|endoftext|>')[0].strip()
|
|
|
|
| 28 |
print(output)
|
| 29 |
```
|
|
|
|
| 25 |
y = model.generate(x, max_length=1024)[0]
|
| 26 |
y_dec = tokenizer.decode(y, max_length=1024, truncation=True)
|
| 27 |
output = y_dec.split('###[Output]:\n')[1].split('<|endoftext|>')[0].strip()
|
| 28 |
+
|
| 29 |
print(output)
|
| 30 |
```
|