mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-01-11 23:00:51 +08:00
bb3b73f947
A simple check for zluda.exe and nccl.dll With hint about AV false positive nccl.dll block.
48 lines
1.1 KiB
Batchfile
48 lines
1.1 KiB
Batchfile
@echo off
|
|
|
|
set FLASH_ATTENTION_TRITON_AMD_ENABLE=TRUE
|
|
set FLASH_ATTENTION_TRITON_AMD_AUTOTUNE=TRUE
|
|
|
|
set MIOPEN_FIND_MODE=2
|
|
set MIOPEN_LOG_LEVEL=3
|
|
|
|
set PYTHON="%~dp0/venv/Scripts/python.exe"
|
|
set GIT=
|
|
set VENV_DIR=./venv
|
|
|
|
set COMMANDLINE_ARGS=--auto-launch --use-quad-cross-attention --reserve-vram 0.9
|
|
|
|
:: in the comfyui-user.bat remove the dots on the line below and change the gfx1030 to your gpu's specific code.
|
|
:: you can find out about yours here, https://llvm.org/docs/AMDGPUUsage.html#processors
|
|
|
|
:: Set TRITON_OVERRIDE_ARCH=gfx1030
|
|
|
|
set ZLUDA_COMGR_LOG_LEVEL=1
|
|
|
|
echo *** Checking and updating to new version if possible
|
|
|
|
copy comfy\customzluda\zluda-default.py comfy\zluda.py /y >NUL
|
|
git pull
|
|
copy comfy\customzluda\zluda.py comfy\zluda.py /y >NUL
|
|
|
|
:: Check for zluda.exe and nccl.dll
|
|
echo.
|
|
echo ** Checking ZLUDA version
|
|
pushd .
|
|
cd ./zluda
|
|
if exist zluda.exe (
|
|
if not exist nccl.dll (
|
|
echo nccl.dll is missing, likely blocked by AV as false positive.
|
|
) else (
|
|
zluda.exe --version
|
|
echo Check passed.
|
|
)
|
|
) else (
|
|
echo Check failed. Can't detect ZLUDA.
|
|
)
|
|
popd
|
|
|
|
echo.
|
|
.\zluda\zluda.exe -- %PYTHON% main.py %COMMANDLINE_ARGS%
|
|
pause
|