Update requirements.txt
Browse files- requirements.txt +3 -10
requirements.txt
CHANGED
@@ -1,16 +1,9 @@
|
|
1 |
-
|
2 |
-
fairscale
|
3 |
-
sentencepiece
|
4 |
-
Pillow
|
5 |
-
huggingface_hub
|
6 |
-
git+https://github.com/csuhan/timm_0_3_2.git
|
7 |
-
git+https://github.com/openai/CLIP.git
|
8 |
-
torch>=2.0.0+cu113
|
9 |
lightning @ git+https://github.com/Lightning-AI/lightning@master
|
|
|
10 |
tqdm # convert_checkpoint.py
|
11 |
numpy # train.py dataset memmap
|
12 |
jsonargparse[signatures] # generate.py, convert_checkpoint.py CLI
|
13 |
bitsandbytes # quantization.py
|
14 |
datasets # evaluate.py
|
15 |
-
zstandard # prepare_redpajama.py
|
16 |
-
git+https://github.com/Lightning-AI/lit-llama.git
|
|
|
1 |
+
torch>=2.0.0
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
2 |
lightning @ git+https://github.com/Lightning-AI/lightning@master
|
3 |
+
sentencepiece
|
4 |
tqdm # convert_checkpoint.py
|
5 |
numpy # train.py dataset memmap
|
6 |
jsonargparse[signatures] # generate.py, convert_checkpoint.py CLI
|
7 |
bitsandbytes # quantization.py
|
8 |
datasets # evaluate.py
|
9 |
+
zstandard # prepare_redpajama.py
|
|