Merge pull request #7 from ThuanNaN/google-colab

Fix: CUDA bug
This commit is contained in:
Nguyen Thuan Duong 2025-01-14 03:58:47 +07:00 committed by GitHub
commit 9a02e5431d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -15,10 +15,11 @@ webdataset==0.2.48
omegaconf==2.3.0
opencv-python==4.7.0.72
decord==0.6.0
peft==0.2.0
peft==0.4.0
sentence-transformers
gradio==3.47.1
accelerate==0.20.3
bitsandbytes==0.37.0
accelerate==0.21.0
bitsandbytes==0.40.2
scikit-image
visual-genome
wandb