PyTorch
English
llama
richardmfan commited on
Commit
90bc69c
·
verified ·
1 Parent(s): c565563

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -0
README.md CHANGED
@@ -37,6 +37,12 @@ outputs = model.generate(**inputs, max_new_tokens=200)
37
  print(tokenizer.decode(outputs[0], skip_special_tokens=True))
38
  ```
39
 
 
 
 
 
 
 
40
  ---
41
 
42
  ## **Evaluation Summary**
 
37
  print(tokenizer.decode(outputs[0], skip_special_tokens=True))
38
  ```
39
 
40
+ ## Using specific checkpoints
41
+ To use a specific checkpoint, use the `revision` argument:
42
+
43
+ ```
44
+ model = AutoModelForCausalLM.from_pretrained("llm360/k2-v2-instruct", device_map="auto", revision="base_0720000")
45
+ ```
46
  ---
47
 
48
  ## **Evaluation Summary**