yindo/rocm-automated
1
0
Fork 0
mirror of https://github.com/BillyOutlast/rocm-automated.git synced 2026-02-04 03:51:19 +01:00
Code Issues Packages Projects Releases Wiki Activity

back to --use-quad-cross-attention

Browse Source
This commit is contained in:
John Doe
2025-11-23 23:52:57 -05:00
parent 60ed1cb3ae
commit f4660f7297
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
comfyui-build/docker-entrypoint.sh
View File

@@ -103,9 +103,9 @@ fi
echo "Using PyTorch index URL: ${PYTORCH_INDEX_URL}"
if echo "${PYTORCH_INDEX_URL}" | grep -q "rocm.nightlies.amd.com"; then
pip install --pre torch torchvision torchaudio pytorch-triton-rocm --extra-index-url ${PYTORCH_INDEX_URL}
pip install --pre torch torchvision torchaudio--extra-index-url ${PYTORCH_INDEX_URL}
else
pip install --pre torch torchvision torchaudio pytorch-triton-rocm --index-url ${PYTORCH_INDEX_URL}
pip install --pre torch torchvision torchaudio --index-url ${PYTORCH_INDEX_URL}
fi
# TO debug flash_attn issues, temporarily disabling its installation
#pip install --upgrade flash_attn --no-build-isolation
@@ -124,7 +124,7 @@ if [ -f "start.sh" ]; then
./start.sh
else
echo "No start.sh found, creating default startup script..."
echo "python main.py --listen 0.0.0.0 --port 8188 --use-pytorch-cross-attention --force-fp16 --disable-xformers" > start.sh
echo "python main.py --listen 0.0.0.0 --port 8188 --use-quad-cross-attention" > start.sh
chmod +x start.sh
./start.sh
fi