Update env

2025-04-09 12:30:45 +00:00 · 2025-01-20 04:30:12 +07:00 · 2025-01-20 04:30:12 +07:00 · fd324283e5
commit fd324283e5
parent 71746e6135
2 changed files with 2 additions and 0 deletions
--- a/environment.yml
+++ b/environment.yml
@ -30,6 +30,7 @@ dependencies:
    - gradio==3.47.1
    - accelerate==0.20.3
    - bitsandbytes==0.37.0
+    - torchmetrics
    - scikit-image
    - visual-genome
    - wandb
--- a/requirements.txt
+++ b/requirements.txt
@ -20,5 +20,6 @@ sentence-transformers
 gradio==3.47.1
 accelerate==0.20.3
 bitsandbytes==0.37.0
+torchmetrics
 scikit-image
 wandb