Chickaboo
/

ChinaLM-9B

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

Chickaboo commited on Jun 18

Commit

5536d19

•

1 Parent(s): 1535c91

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -56,8 +56,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 device = "cuda" # the device to load the model onto
-model = AutoModelForCausalLM.from_pretrained("Chickaboo/ChinaLM")
-tokenizer = AutoTokenizer.from_pretrained("Chickaboo/ChinaLM")
 messages = [
     {"role": "user", "content": "What is your favourite condiment?"},

 device = "cuda" # the device to load the model onto
+model = AutoModelForCausalLM.from_pretrained("Chickaboo/ChinaLM-9B")
+tokenizer = AutoTokenizer.from_pretrained("Chickaboo/ChinaLM-9B")
 messages = [
     {"role": "user", "content": "What is your favourite condiment?"},