From ff933b756383053dd18a10a977e229d85313d1e7 Mon Sep 17 00:00:00 2001 From: Zernel Date: Tue, 7 Jan 2025 15:30:28 +0800 Subject: [PATCH] Update requirements --- requirements.txt | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/requirements.txt b/requirements.txt index 94c9497..52ef9b8 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,5 +2,9 @@ numpy Pillow Requests flash_attn -transformers -einops \ No newline at end of file +accelerate>=0.25.0 +huggingface-hub +transformers>=4.44.0 +einops>=0.7.0 +torch>=2.1.2 +torchvision>=0.16.2 \ No newline at end of file