This commit is contained in:
ThuanNaN 2025-01-20 05:24:55 +07:00
parent fd324283e5
commit 97601f7a90
2 changed files with 4 additions and 2 deletions

View File

@ -30,7 +30,8 @@ dependencies:
- gradio==3.47.1 - gradio==3.47.1
- accelerate==0.20.3 - accelerate==0.20.3
- bitsandbytes==0.37.0 - bitsandbytes==0.37.0
- torchmetrics - pycocotools
- torchmetrics[detection]
- scikit-image - scikit-image
- visual-genome - visual-genome
- wandb - wandb

View File

@ -20,6 +20,7 @@ sentence-transformers
gradio==3.47.1 gradio==3.47.1
accelerate==0.20.3 accelerate==0.20.3
bitsandbytes==0.37.0 bitsandbytes==0.37.0
torchmetrics pycocotools
torchmetrics[detection]
scikit-image scikit-image
wandb wandb