From 0474451a6a952feee084b7f11460d64be7a26b02 Mon Sep 17 00:00:00 2001 From: Daniel Han-Chen Date: Thu, 25 Jan 2024 03:55:26 +1100 Subject: [PATCH] Update pyproject.toml --- pyproject.toml | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) 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",