fix: env

2025-04-04 18:10:47 +00:00 · 2025-01-20 05:24:55 +07:00 · 2025-01-20 05:24:55 +07:00 · 97601f7a90
commit 97601f7a90
parent fd324283e5
2 changed files with 4 additions and 2 deletions
--- a/environment.yml
+++ b/environment.yml
@ -30,7 +30,8 @@ dependencies:
    - gradio==3.47.1
    - accelerate==0.20.3
    - bitsandbytes==0.37.0
-    - torchmetrics
+    - pycocotools
+    - torchmetrics[detection]
    - scikit-image
    - visual-genome
    - wandb
--- a/requirements.txt
+++ b/requirements.txt
@ -20,6 +20,7 @@ sentence-transformers
 gradio==3.47.1
 accelerate==0.20.3
 bitsandbytes==0.37.0
-torchmetrics
+pycocotools
+torchmetrics[detection]
 scikit-image
 wandb