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
|