PY007 commited on
Commit
b446322
1 Parent(s): 213ebf6

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -32,7 +32,7 @@ Do check the [TinyLlama](https://github.com/jzhang38/TinyLlama) github page for
32
  from transformers import AutoTokenizer
33
  import transformers
34
  import torch
35
- model = "PY007/TinyLlama-1.1B-step-50K-105b"
36
  tokenizer = AutoTokenizer.from_pretrained(model)
37
  pipeline = transformers.pipeline(
38
  "text-generation",
 
32
  from transformers import AutoTokenizer
33
  import transformers
34
  import torch
35
+ model = "PY007/TinyLlama-1.1B-intermediate-step-240k-503b"
36
  tokenizer = AutoTokenizer.from_pretrained(model)
37
  pipeline = transformers.pipeline(
38
  "text-generation",