Jonathan Tow commited on
Commit
e2b758d
1 Parent(s): fe16172

fix: lower-case org name in example `from_pretrained`

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