Text Generation
Transformers
PyTorch
Safetensors
English
llama
text-generation-inference
Inference Endpoints
Sefika commited on
Commit
edcd968
1 Parent(s): 1ea9349

Update README.md

Browse files

edit model name and add "python" into script.

Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -25,11 +25,11 @@ The dataset used is [OpenAssistant/oasst_top1_2023-08-25](https://huggingface.co
25
  #### How to use
26
  You will need the transformers>=4.31
27
  Do check the [TinyLlama](https://github.com/jzhang38/TinyLlama) github page for more information.
28
- ```
29
  from transformers import AutoTokenizer
30
  import transformers
31
  import torch
32
- model = "PY007/TinyLlama-1.1B-Chat-v0.3"
33
  tokenizer = AutoTokenizer.from_pretrained(model)
34
  pipeline = transformers.pipeline(
35
  "text-generation",
 
25
  #### How to use
26
  You will need the transformers>=4.31
27
  Do check the [TinyLlama](https://github.com/jzhang38/TinyLlama) github page for more information.
28
+ ```python
29
  from transformers import AutoTokenizer
30
  import transformers
31
  import torch
32
+ model = "TinyLlama/TinyLlama-1.1B-Chat-v0.3"
33
  tokenizer = AutoTokenizer.from_pretrained(model)
34
  pipeline = transformers.pipeline(
35
  "text-generation",