diff --git a/src/llmtuner/model/loader.py b/src/llmtuner/model/loader.py index 234356ad..404da42a 100644 --- a/src/llmtuner/model/loader.py +++ b/src/llmtuner/model/loader.py @@ -20,7 +20,7 @@ if TYPE_CHECKING: logger = get_logger(__name__) -require_version("transformers>=4.36.1", "To fix: pip install transformers>=4.36.1") +require_version("transformers>=4.36.2", "To fix: pip install transformers>=4.36.2") 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.7.0", "To fix: pip install peft>=0.7.0")