hardmaru commited on
Commit
d4332bd
1 Parent(s): 3eda748

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -27,8 +27,8 @@ Get started chatting with `StableLM-Tuned-Alpha` by using the following code sni
27
  ```python
28
  from transformers import AutoModelForCausalLM, AutoTokenizer
29
 
30
- tokenizer = AutoTokenizer.from_pretrained("StabilityAI/stablelm-tuned-alpha-3b")
31
- model = AutoModelForCausalLM.from_pretrained("StabilityAI/stablelm-tuned-alpha-3b")
32
  model.half().cuda()
33
 
34
  system_prompt = """<|SYSTEM|># StableLM Tuned (Alpha version)
 
27
  ```python
28
  from transformers import AutoModelForCausalLM, AutoTokenizer
29
 
30
+ tokenizer = AutoTokenizer.from_pretrained("StabilityAI/stablelm-tuned-alpha-7b")
31
+ model = AutoModelForCausalLM.from_pretrained("StabilityAI/stablelm-tuned-alpha-7b")
32
  model.half().cuda()
33
 
34
  system_prompt = """<|SYSTEM|># StableLM Tuned (Alpha version)