Felladrin commited on
Commit
5a95afc
1 Parent(s): f14887d

Update model name to use in transformers

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-intermediate-step-240k-503b"
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-715k-1.5T"
36
  tokenizer = AutoTokenizer.from_pretrained(model)
37
  pipeline = transformers.pipeline(
38
  "text-generation",