Update README.md
Browse files
README.md
CHANGED
|
@@ -47,5 +47,4 @@ response = tokenizer.decode(outputs[0])
|
|
| 47 |
outputs = model.generate(inputs=input_ids.cuda(), attention_mask=attention_mask.cuda(), max_new_tokens=10)
|
| 48 |
print(tokenizer.decode(outputs[0]))
|
| 49 |
```
|
| 50 |
-
```
|
| 51 |
|
|
|
|
| 47 |
outputs = model.generate(inputs=input_ids.cuda(), attention_mask=attention_mask.cuda(), max_new_tokens=10)
|
| 48 |
print(tokenizer.decode(outputs[0]))
|
| 49 |
```
|
|
|
|
| 50 |
|