stabilityai
/

stablelm-tuned-alpha-3b

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

reshinthadith commited on Apr 19, 2023

Commit

56f0d9b

·

1 Parent(s): f7257a9

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -26,8 +26,8 @@ Get started chatting with `StableLM-Tuned-Alpha` by using the following code sni
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
-tokenizer = AutoTokenizer.from_pretrained("StabilityAI/stablelm-tuned-alpha-7b")
-model = AutoModelForCausalLM.from_pretrained("StabilityAI/stablelm-tuned-alpha-7b")
 model.half().cuda()
 inputs = tokenizer("What's your mood today?", return_tensors="pt").to("cuda")

 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
+tokenizer = AutoTokenizer.from_pretrained("StabilityAI/stablelm-tuned-alpha-3b")
+model = AutoModelForCausalLM.from_pretrained("StabilityAI/stablelm-tuned-alpha-3b")
 model.half().cuda()
 inputs = tokenizer("What's your mood today?", return_tensors="pt").to("cuda")