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