2121-8
/

TinySlime-1.1B-Chat-v1.0

Text Generation

OpenAccess AI Collective

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

2121-8 commited on Oct 1

Commit

4195939

•

1 Parent(s): 18f8085

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -60,7 +60,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 model_name = "2121-8/TinySlime-1.1B-Chat-v1.0"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModel.from_pretrained(model_name)
 ```
 ### テキスト生成

 model_name = "2121-8/TinySlime-1.1B-Chat-v1.0"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)
 ```
 ### テキスト生成