diff --git a/pyproject.toml b/pyproject.toml index 00be056a99..49249e952f 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -129,7 +129,7 @@ kaggle = [ "unsloth[huggingface]", ] kaggle-new = [ - "xformers", + "xformers @ https://download.pytorch.org/whl/cu121", "unsloth[huggingface]", ] conda = [