diff --git a/pyproject.toml b/pyproject.toml index 0134a30..dfcaf9b 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -13,7 +13,7 @@ train = [ "unsloth[colab-new]>=2024.8", "transformers>=4.40", "trl>=0.8", - "torch>=2.0", + "torch>=2.11.0", "bitsandbytes>=0.43", ] dev = [