Update README.md
Browse files
README.md
CHANGED
|
@@ -49,6 +49,7 @@ dtype: float16
|
|
| 49 |
|
| 50 |
### How to Use
|
| 51 |
|
|
|
|
| 52 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 53 |
import torch
|
| 54 |
tokenizer = AutoTokenizer.from_pretrained("m-a-p/OpenCodeInterpreter-DS-6.7B")
|
|
@@ -67,3 +68,5 @@ outputs = model.generate(
|
|
| 67 |
eos_token_id=tokenizer.eos_token_id,
|
| 68 |
)
|
| 69 |
print(tokenizer.decode(outputs[0][len(inputs[0]):], skip_special_tokens=True))
|
|
|
|
|
|
|
|
|
| 49 |
|
| 50 |
### How to Use
|
| 51 |
|
| 52 |
+
```
|
| 53 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 54 |
import torch
|
| 55 |
tokenizer = AutoTokenizer.from_pretrained("m-a-p/OpenCodeInterpreter-DS-6.7B")
|
|
|
|
| 68 |
eos_token_id=tokenizer.eos_token_id,
|
| 69 |
)
|
| 70 |
print(tokenizer.decode(outputs[0][len(inputs[0]):], skip_special_tokens=True))
|
| 71 |
+
|
| 72 |
+
```
|