lingyuzeng/cdc_dockerfile
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
lingyuzeng
2024-07-16 13:33:51 +08:00
parent d288ed7214
commit 5ab69cead0
1 changed files with 37 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

74
finetune/requirements.txt
View File

@@ -1,37 +1,37 @@
pytorch pytorch
torchvision torchvision
torchaudio torchaudio
pydantic pydantic
transformers transformers
datasets datasets
accelerate accelerate
evaluate evaluate
peft peft
deepspeed deepspeed
tiktoken tiktoken
sentencepiece sentencepiece
tqdm tqdm
nltk nltk
matplotlib matplotlib
seaborn seaborn
numpy numpy
pandas pandas
scikit-learn scikit-learn
diffusers diffusers
huggingface_hub huggingface_hub
spacy spacy
Pillow Pillow
blobfile blobfile
requests requests
scipy scipy
pycocotools pycocotools
protobuf protobuf
timm timm
pyyaml pyyaml
ipython ipython
xformers xformers
opencv-contrib-python opencv-contrib-python
open_clip_torch open_clip_torch
flash-attn flash-attn
packaging packaging
psutil psutil