From c4a59b0380144bfd689b28700a79dedea17433f7 Mon Sep 17 00:00:00 2001 From: ver217 Date: Thu, 6 Jun 2024 14:11:52 +0800 Subject: [PATCH] [misc] update requirements --- requirements/requirements.txt | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/requirements/requirements.txt b/requirements/requirements.txt index d30b26dbc787..fa88501ef968 100644 --- a/requirements/requirements.txt +++ b/requirements/requirements.txt @@ -8,7 +8,7 @@ click fabric contexttimer ninja -torch>=2.1.0 +torch>=2.1.0,<2.3.0 safetensors einops pydantic @@ -16,7 +16,7 @@ ray sentencepiece google protobuf -transformers==4.36.2 +transformers>=4.36.2,<4.40.0 peft>=0.7.1 bitsandbytes>=0.39.0 rpyc==6.0.0