Skip to content

Commit 953636e

Browse files
authored
Update README.md
1 parent ccedd77 commit 953636e

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

README.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
1. [Quickstart](#quickstart)
88
- [Installation](#installation)
99
- [Code generation with StarCoder](#code-generation)
10-
- [Text-generation-inference code](#text-generation-inference-code)
10+
- [Text-generation-inference code](#text-generation-inference)
1111
2. [Fine-tuning](#fine-tuning)
1212
- [Step by step installation with conda](#step-by-step-installation-with-conda)
1313
- [Datasets](#datasets)
@@ -50,7 +50,7 @@ pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, device=0)
5050
print( pipe("def hello():") )
5151
```
5252

53-
## Text-generation-inference Code
53+
## Text-generation-inference
5454

5555
```bash
5656
docker run --gpus '"device:0"' -p 8080:80 -v $PWD/data:/data -e HUGGING_FACE_HUB_TOKEN=<YOUR BIGCODE ENABLED TOKEN> -e HF_HUB_ENABLE_HF_TRANSFER=0 -d ghcr.io/huggingface/text-generation-inference:sha-880a76e --model-id bigcode/starcoder --max-total-tokens 8192
@@ -75,7 +75,7 @@ conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cud
7575
Install `transformers` and `peft`
7676
```bash
7777
conda install -c huggingface transformers
78-
pip install git+https://github.com/huggingface/peft.git
78+
pip install git+https://github.com/huggingface/peft.git
7979
```
8080
Note that you can install the latest stable version of transformers by using
8181

0 commit comments

Comments
 (0)