diff --git a/pyproject.toml b/pyproject.toml index ea420e2ffb..506e9a7552 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -98,10 +98,14 @@ colab_ampere = [ "flash-attn", ] colab_torch211 = [ - "unsloth[cu121_torch211]", + "unsloth[huggingface]", + "bitsandbytes", + "unsloth[cu121onlytorch211]", ] colab_ampere_torch211 = [ - "unsloth[cu121_torch211]", + "unsloth[huggingface]", + "bitsandbytes", + "unsloth[cu121onlytorch211]", "packaging", "ninja", "flash-attn",