TinyLlama
/

TinyLlama-1.1B-intermediate-step-955k-token-2T

Text Generation

text-generation-inference

Model card Files Files and versions

PY007 commited on Nov 19, 2023

Commit

ee7a31d

·

1 Parent(s): b2be58a

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -47,7 +47,7 @@ Do check the [TinyLlama](https://github.com/jzhang38/TinyLlama) github page for
 from transformers import AutoTokenizer
 import transformers
 import torch
-model = "PY007/TinyLlama-1.1B-step-50K-105b"
 tokenizer = AutoTokenizer.from_pretrained(model)
 pipeline = transformers.pipeline(
     "text-generation",

 from transformers import AutoTokenizer
 import transformers
 import torch
+model = "TinyLlama/TinyLlama-1.1B-intermediate-step-955k-token-2T"
 tokenizer = AutoTokenizer.from_pretrained(model)
 pipeline = transformers.pipeline(
     "text-generation",