様々な LoRAを作成することのできるAI-Toolkit

set HF_HOME=E:\my_cache\hf_home
set MODELSCOPE_CACHE=E:\my_cache\MODELSCOPE
set TRANSFORMERS_CACHE=E:\my_cache\transformers
set DIFFUSERS_CACHE=E:\my_cache\diffusers
echo HF_HOME = %HF_HOME%
echo TRANSFORMERS_CACHE = %TRANSFORMERS_CACHE%
echo DIFFUSERS_CACHE = %DIFFUSERS_CACHE%

@echo off&&cd /d %~dp0
Title AI-Toolkit-Easy-Install v0.3.30 by ivo
setlocal enabledelayedexpansion

set PYTHONPATH=
set PYTHONHOME=
set PYTHON=
set PYTHONSTARTUP=
set PYTHONUSERBASE=
set PIP_CONFIG_FILE=
set PIP_REQUIRE_VIRTUALENV=
set VIRTUAL_ENV=
set CONDA_PREFIX=
set CONDA_DEFAULT_ENV=
set PYENV_ROOT=
set PYENV_VERSION=
set HF_HOME=E:\my_cache\hf_home
set MODELSCOPE_CACHE=E:\my_cache\MODELSCOPE
set TRANSFORMERS_CACHE=E:\my_cache\transformers
set DIFFUSERS_CACHE=E:\my_cache\diffusers
echo HF_HOME = %HF_HOME%
echo TRANSFORMERS_CACHE = %TRANSFORMERS_CACHE%
echo DIFFUSERS_CACHE = %DIFFUSERS_CACHE%
echo Cleaning up any stuck processes...

REM Set ports
set UI_PORT=8675
set WORKER_PORT=8675


REM Function to kill processes on a specific port
call :KillPort %UI_PORT%
call :KillPort %WORKER_PORT%

REM Wait for ports to be fully released
timeout /t 3 >nul

echo.

set GIT_LFS_SKIP_SMUDGE=1
set "local_serv=http://localhost:8675"
echo.
cd ./ai-toolkit
    echo ::::::::::::  Starting AI-Toolkit ...  ::::::::::::::
    echo.
git.exe fetch
git.exe status -uno | findstr /C:"Your branch is behind" >nul
if !errorlevel!==0 (
    echo :::::::::::: New updates are available ::::::::::::::
    echo :::::::::::: Run Update-AI-Toolkit.bat ::::::::::::::
    echo.
)

echo Tips for beginners:
echo.
echo General:
echo  1. Set your Hugging Face Token in Settings
echo  2. Close server with Ctrl+C twice, not the X button
echo  3. To activate the virtual environment (if needed):
echo     - Open CMD where Start-AI-Toolkit.bat is located
echo       and Run AI-Toolkit\venv\Scripts\activate.bat
echo     OR just start venv-AI-Toolkit.bat
echo.
echo :::::::: Waiting for the server to start... :::::::::

cd ./ui
if  exist .\node_modules\.prisma\client\query_engine-windows.dll.node (
copy /Y .\node_modules\.prisma\client\query_engine-windows.dll.node .\node_modules\.prisma\client\query_engine-windows.dll.node.k
del/q .\node_modules\.prisma\client\query_engine-windows.dll.node 

if  exist .\node_modules\.prisma\client\query_engine-windows.dll.node (
for /f "tokens=2" %%p in ('tasklist ^| findstr /i "node.exe"') do (
        echo   - Found stuck Node.js process %%p, terminating...
        taskkill /PID %%p /F >nul 2>&1
    )
)


)

start cmd.exe /k npm run build_and_start
:loop
if exist "%windir%\System32\WindowsPowerShell\v1.0" set "PATH=%PATH%;%windir%\System32\WindowsPowerShell\v1.0"
powershell -Command "try { $response = Invoke-WebRequest -Uri '!local_serv!' -TimeoutSec 2 -UseBasicParsing; exit 0 } catch { exit 1 }" >nul 2>&1
if !errorlevel! neq 0 (timeout /t 2 /nobreak >nul&&goto :loop)
start !local_serv!

REM Cleanup after processes end
echo.
echo Processes ended. Cleaning up...
call :KillPort %UI_PORT%
call :KillPort %WORKER_PORT%

pause
goto :eof

REM ============================================
REM Subroutine to kill processes on a port
REM ============================================
:KillPort
set TARGET_PORT=%1
echo Checking port %TARGET_PORT%...

REM Find and kill any process listening on the port (both IPv4 and IPv6)
for /f "tokens=5" %%p in ('netstat -ano ^| findstr ":%TARGET_PORT% " ^| findstr LISTENING') do (
    echo   - Killing process %%p on port %TARGET_PORT%
    taskkill /PID %%p /F >nul 2>&1
    if not errorlevel 1 (
        echo   - Process %%p terminated successfully
    )
)

REM Also check for Node.js processes that might be stuck
for /f "tokens=2" %%p in ('tasklist ^| findstr /i "node.exe"') do (
    netstat -ano | findstr "%%p" | findstr ":%TARGET_PORT% " | findstr LISTENING >nul
    if not errorlevel 1 (
        echo   - Found stuck Node.js process %%p, terminating...
        taskkill /PID %%p /F >nul 2>&1
    )
)

goto :eof
REM Cleanup after processes end
echo.
echo Processes ended. Cleaning up...
call :KillPort %UI_PORT%
call :KillPort %WORKER_PORT%

pause
goto :eof