richardmfan commited on
Commit
fdb875f
·
verified ·
1 Parent(s): 56dd218

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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**