mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-01-11 06:40:48 +08:00
53 lines
1.5 KiB
Batchfile
53 lines
1.5 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=%~dp0venv\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 a new version if possible...
|
|
copy comfy\customzluda\zluda-default.py comfy\zluda.py /y >NUL
|
|
setlocal EnableDelayedExpansion
|
|
for /f "delims=" %%A in ('git pull 2^>^&1') do (
|
|
echo ** %%A
|
|
)
|
|
endlocal
|
|
copy comfy\customzluda\zluda.py comfy\zluda.py /y >NUL
|
|
|
|
rem Check for zluda.exe and nccl.dll inside the zluda folder
|
|
echo.
|
|
echo ** Checking ZLUDA installation...
|
|
pushd .\zluda
|
|
if exist zluda.exe (
|
|
if not exist nccl.dll (
|
|
for /f "tokens=2 delims= " %%v in ('zluda.exe --version 2^>^&1') do (
|
|
echo ** [FAIL] Detected version [%%v], but nccl.dll is missing. Likely blocked by AV as false positive.
|
|
)
|
|
) else (
|
|
for /f "tokens=2 delims= " %%v in ('zluda.exe --version 2^>^&1') do (
|
|
echo ** [PASS] Detected version: %%v
|
|
)
|
|
)
|
|
) else (
|
|
echo ** [FAIL] Can't detect zluda.exe inside .\zluda directory.
|
|
)
|
|
popd
|
|
|
|
echo.
|
|
.\zluda\zluda.exe -- %PYTHON% main.py %COMMANDLINE_ARGS%
|
|
pause
|