diff --git a/pyproject.toml b/pyproject.toml index 4ff6534dba..3ea50ca00e 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -161,7 +161,7 @@ colab-ampere-torch220 = [ "ninja", "flash-attn", ] -colab-1 = [ +colab-new = [ "tyro", "transformers>=4.38.2", "datasets>=2.16.0", @@ -171,22 +171,13 @@ colab-1 = [ "wheel>=0.42.0", "numpy", ] -colab-2 = [ +colab-no-deps = [ "accelerate>=0.26.1", "trl>=0.7.9", "peft>=0.7.1", "xformers", "bitsandbytes", ] -colab-ampere-1 = [ - "unsloth[colab-1]", -] -colab-ampere-2 = [ - "packaging", - "ninja", - "flash-attn", - "unsloth[colab-2]", -] colab-ampere = [ "unsloth[colab-ampere-torch220]", "packaging",