diff --git a/requirements.txt b/requirements.txt index ca876344..bbd1f89d 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,8 +1,8 @@ torch>=1.13.1 transformers>=4.31.0,<4.35.0 -datasets>=2.14.0 +datasets>=2.14.3 accelerate>=0.21.0 -peft>=0.6.0 +peft==0.6.0 trl>=0.7.4 gradio>=3.38.0,<4.0.0 scipy diff --git a/src/llmtuner/model/loader.py b/src/llmtuner/model/loader.py index faba1ee2..728472da 100644 --- a/src/llmtuner/model/loader.py +++ b/src/llmtuner/model/loader.py @@ -39,9 +39,9 @@ logger = get_logger(__name__) require_version("transformers>=4.31.0,<4.35.0", "To fix: pip install \"transformers>=4.31.0,<4.35.0\"") -require_version("datasets>=2.14.0", "To fix: pip install datasets>=2.14.0") +require_version("datasets>=2.14.3", "To fix: pip install datasets>=2.14.3") require_version("accelerate>=0.21.0", "To fix: pip install accelerate>=0.21.0") -require_version("peft>=0.6.0", "To fix: pip install peft>=0.6.0") +require_version("peft==0.6.0", "To fix: pip install peft==0.6.0") require_version("trl>=0.7.4", "To fix: pip install trl>=0.7.4")