stabilityai
/

stablelm-tuned-alpha-3b

Text Generation

text-generation-inference

Model card Files Files and versions

Update README.md

#6

by vishank97 - opened Apr 25, 2023

base: refs/heads/main

←

from: refs/pr/6

Discussion Files changed

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -27,8 +27,8 @@ Get started chatting with `StableLM-Tuned-Alpha` by using the following code sni
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer, StoppingCriteria, StoppingCriteriaList
-tokenizer = AutoTokenizer.from_pretrained("StabilityAI/stablelm-tuned-alpha-7b")
-model = AutoModelForCausalLM.from_pretrained("StabilityAI/stablelm-tuned-alpha-7b")
 model.half().cuda()
 class StopOnTokens(StoppingCriteria):

 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer, StoppingCriteria, StoppingCriteriaList
+tokenizer = AutoTokenizer.from_pretrained("StabilityAI/stablelm-tuned-alpha-3b")
+model = AutoModelForCausalLM.from_pretrained("StabilityAI/stablelm-tuned-alpha-3b")
 model.half().cuda()
 class StopOnTokens(StoppingCriteria):