Update README.md
Browse files
README.md
CHANGED
|
@@ -83,7 +83,7 @@ To use a specific checkpoint, use the `revision` argument:
|
|
| 83 |
|
| 84 |
**Using transformers**
|
| 85 |
```
|
| 86 |
-
model = AutoModelForCausalLM.from_pretrained("llm360/k2-v2-instruct", device_map="auto", revision=stage_1_0135000)
|
| 87 |
```
|
| 88 |
|
| 89 |
**Using vLLM**
|
|
|
|
| 83 |
|
| 84 |
**Using transformers**
|
| 85 |
```
|
| 86 |
+
model = AutoModelForCausalLM.from_pretrained("llm360/k2-v2-instruct", device_map="auto", revision="stage_1_0135000")
|
| 87 |
```
|
| 88 |
|
| 89 |
**Using vLLM**
|