diff --git a/torch.js b/torch.js index cb7d3fc..ce571dd 100644 --- a/torch.js +++ b/torch.js @@ -23,7 +23,7 @@ module.exports = { "venv": "{{args && args.venv ? args.venv : null}}", "path": "{{args && args.path ? args.path : '.'}}", "message": [ - "uv pip install torch==2.7.0 torchvision==0.22.0 torchaudio==2.7.0 {{args && args.xformers ? 'xformers==0.0.30' : ''}} --index-url https://download.pytorch.org/whl/cu128 --force-reinstall", + "uv pip install torch==2.7.0 torchvision==0.22.0 torchaudio==2.7.0 --index-url https://download.pytorch.org/whl/cu128 --force-reinstall", "uv pip install https://huggingface.co/MonsterMMORPG/SECourses_Premium_Flash_Attention/resolve/main/sageattention-2.1.1-cp310-cp310-linux_x86_64.whl", "uv pip install https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.7.16/flash_attn-2.7.4+cu128torch2.7-cp310-cp310-linux_x86_64.whl", "uv pip install numpy==2.1.2"