Update README.md
Browse files
README.md
CHANGED
|
@@ -66,7 +66,7 @@ Detailed performance and introduction are shown in our <a href="https://arxiv.or
|
|
| 66 |
```python
|
| 67 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 68 |
|
| 69 |
-
model_name = "
|
| 70 |
device = "cuda"
|
| 71 |
|
| 72 |
model = AutoModelForCausalLM.from_pretrained(
|
|
|
|
| 66 |
```python
|
| 67 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 68 |
|
| 69 |
+
model_name = "BitStarWalkin/SuperCorrect-7B"
|
| 70 |
device = "cuda"
|
| 71 |
|
| 72 |
model = AutoModelForCausalLM.from_pretrained(
|