Update README.md
Browse files
README.md
CHANGED
|
@@ -150,7 +150,7 @@ trainer.train()
|
|
| 150 |
|
| 151 |
### Outputs by REILX/Phi-3-medium-128k-code-instruct
|
| 152 |
```python
|
| 153 |
-
|
| 154 |
from transformers import RobertaForCausalLM, RobertaTokenizer
|
| 155 |
|
| 156 |
# Load the model and tokenizer
|
|
|
|
| 150 |
|
| 151 |
### Outputs by REILX/Phi-3-medium-128k-code-instruct
|
| 152 |
```python
|
| 153 |
+
import torch
|
| 154 |
from transformers import RobertaForCausalLM, RobertaTokenizer
|
| 155 |
|
| 156 |
# Load the model and tokenizer
|