stabilityai
/

stablelm-tuned-alpha-3b

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

log0 commited on Apr 26, 2023

Commit

45df53b

•

1 Parent(s): d1c03d2

Update README.md

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):