From 9d18dac9baa249df22b83afd36d5aa796b031970 Mon Sep 17 00:00:00 2001 From: moon7star9 Date: Mon, 10 Feb 2025 15:08:12 +0800 Subject: [PATCH] update requirements.txt --- requirements.txt | 19 ++++++++++++++++++- 1 file changed, 18 insertions(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index 17564df..2b5e4c5 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1 +1,18 @@ -transformers>=4.43.4 \ No newline at end of file +transformers + +torch>=2.5.0 +torchvision>=0.20.0 +numpy<2 +opencv-python +timm +scipy +scikit-image +kornia +einops + +tqdm +prettytable + +transformers<=4.48.0 +huggingface-hub>0.25 +accelerate \ No newline at end of file