Sefika commited on
Commit
d3dc6e6
1 Parent(s): ae666e2

Update README.md

Browse files

change model as "TinyLlama/TinyLlama-1.1B-step-50K-105b" and add "python"

Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -43,11 +43,11 @@ We will be rolling out intermediate checkpoints following the below schedule. We
43
  #### How to use
44
  You will need the transformers>=4.31
45
  Do check the [TinyLlama](https://github.com/jzhang38/TinyLlama) github page for more information.
46
- ```
47
  from transformers import AutoTokenizer
48
  import transformers
49
  import torch
50
- model = "PY007/TinyLlama-1.1B-step-50K-105b"
51
  tokenizer = AutoTokenizer.from_pretrained(model)
52
  pipeline = transformers.pipeline(
53
  "text-generation",
 
43
  #### How to use
44
  You will need the transformers>=4.31
45
  Do check the [TinyLlama](https://github.com/jzhang38/TinyLlama) github page for more information.
46
+ ```python
47
  from transformers import AutoTokenizer
48
  import transformers
49
  import torch
50
+ model = "TinyLlama/TinyLlama-1.1B-step-50K-105b"
51
  tokenizer = AutoTokenizer.from_pretrained(model)
52
  pipeline = transformers.pipeline(
53
  "text-generation",