Merge pull request #5 from ThuanNaN/google-colab

add requirements.txt
This commit is contained in:
Nguyen Thuan Duong 2025-01-13 23:07:27 +07:00 committed by GitHub
commit 09fd7cb6e7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

24
requirements.txt Normal file
View File

@ -0,0 +1,24 @@
torch==2.0.0
torchaudio
torchvision
huggingface-hub==0.18.0
matplotlib==3.7.0
psutil==5.9.4
iopath
pyyaml==6.0
regex==2022.10.31
tokenizers==0.13.2
tqdm==4.64.1
transformers==4.30.0
timm==0.6.13
webdataset==0.2.48
omegaconf==2.3.0
opencv-python==4.7.0.72
decord==0.6.0
peft==0.2.0
sentence-transformers
gradio==3.47.1
accelerate==0.20.3
bitsandbytes==0.37.0
scikit-image
wandb