xuxw98 commited on
Commit
bc1ae16
·
1 Parent(s): 9099de4

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -10
requirements.txt CHANGED
@@ -1,16 +1,9 @@
1
- --extra-index-url https://download.pytorch.org/whl/cu113
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