reshinthadith commited on
Commit
cb246fb
1 Parent(s): 881125b

Updated model identifier in the example code.

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -19,8 +19,8 @@ Get started generating text with `StableLM-Base-Alpha` by using the following co
19
  ```python
20
  from transformers import AutoModelForCausalLM, AutoTokenizer
21
 
22
- tokenizer = AutoTokenizer.from_pretrained("StabilityAI/stablelm-base-alpha-7b")
23
- model = AutoModelForCausalLM.from_pretrained("StabilityAI/stablelm-base-alpha-7b")
24
  model.half().cuda()
25
 
26
  inputs = tokenizer("What's your mood today?", return_tensors="pt").to("cuda")
 
19
  ```python
20
  from transformers import AutoModelForCausalLM, AutoTokenizer
21
 
22
+ tokenizer = AutoTokenizer.from_pretrained("StabilityAI/stablelm-base-alpha-3b")
23
+ model = AutoModelForCausalLM.from_pretrained("StabilityAI/stablelm-base-alpha-3b")
24
  model.half().cuda()
25
 
26
  inputs = tokenizer("What's your mood today?", return_tensors="pt").to("cuda")