diff --git a/pyproject.toml b/pyproject.toml index 8b1a6b67a2..9abe7a5d88 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -299,6 +299,7 @@ colab-new = [ "huggingface_hub", "hf_transfer", "bitsandbytes>=0.43.3", + "unsloth[triton]", ] colab-no-deps = [ "accelerate>=0.34.1",