From 97601f7a90d2a2774941144ab2585bcc2cae8504 Mon Sep 17 00:00:00 2001 From: ThuanNaN Date: Mon, 20 Jan 2025 05:24:55 +0700 Subject: [PATCH] fix: env --- environment.yml | 3 ++- requirements.txt | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/environment.yml b/environment.yml index e6b6071..44774b5 100644 --- 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 diff --git a/requirements.txt b/requirements.txt index 063befe..4a3376a 100644 --- 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 \ No newline at end of file