TinyLlama
/

TinyLlama-1.1B-intermediate-step-240k-503b

Text Generation

text-generation-inference

Model card Files Files and versions

PY007 commited on Sep 19, 2023

Commit

b446322

·

1 Parent(s): 213ebf6

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -32,7 +32,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 = "PY007/TinyLlama-1.1B-intermediate-step-240k-503b"
 tokenizer = AutoTokenizer.from_pretrained(model)
 pipeline = transformers.pipeline(
     "text-generation",