PY007 Felladrin commited on
Commit
1fa3971
1 Parent(s): f14887d

Update model name in the code snippet (#1)

Browse files

- Update model name to use in transformers (5a95afcf4a1e71f55db9722720bf27bfd624cf40)


Co-authored-by: Victor Nogueira <[email protected]>

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",