a | b/requirements.txt | ||
---|---|---|---|
1 | accelerate==0.18.0 |
||
2 | loralib==0.1.1 |
||
3 | bitsandbytes==0.38.1 |
||
4 | datasets==2.11.0 |
||
5 | fire==0.5.0 |
||
6 | sentencepiece==0.1.98 |
||
7 | gradio==3.27.0 |
||
8 | iterative-stratification==0.1.7 |
||
9 | iopath==0.1.10 |
||
10 | omegaconf==2.3.0 |
||
11 | packaging==21.3 |
||
12 | pandas==1.3.5 |
||
13 | pycocoevalcap==1.2 |
||
14 | scikit-image==0.18.1 |
||
15 | spacy==3.5.1 |
||
16 | timm==0.4.12 |
||
17 | torch==1.13.0 |
||
18 | torchvision==0.14.0 |
||
19 | tqdm==4.65.0 |
||
20 | peft @ git+https://github.com/huggingface/peft.git@e536616888d51b453ed354a6f1e243fecb02ea08 |
||
21 | numpy==1.21.6 |
||
22 | wandb==0.14.0 |
||
23 | pytorch_lightning==1.6.5 |
||
24 | torchinfo==1.8.0 |
||
25 | nltk==3.8.1 |