mirror of https://github.com/bmaltais/kohya_ss
32 lines
912 B
Batchfile
32 lines
912 B
Batchfile
@echo off
|
|
|
|
set PYTHON_VER=3.10.9
|
|
|
|
:: Deactivate the virtual environment
|
|
call .\venv\Scripts\deactivate.bat
|
|
|
|
:: Activate the virtual environment
|
|
call .\venv\Scripts\activate.bat
|
|
set PATH=%PATH%;%~dp0venv\Lib\site-packages\torch\lib
|
|
|
|
:: If the first argument is --help, skip the validation step
|
|
if "%~1" equ "--help" goto :skip_validation
|
|
|
|
:: Validate requirements
|
|
python.exe .\setup\validate_requirements.py
|
|
if %errorlevel% neq 0 exit /b %errorlevel%
|
|
|
|
:skip_validation
|
|
|
|
:: If the exit code is 0, run the kohya_gui.py script with the command-line arguments
|
|
if %errorlevel% equ 0 (
|
|
REM Check if the batch was started via double-click
|
|
IF /i "%comspec% /c %~0 " equ "%cmdcmdline:"=%" (
|
|
REM echo This script was started by double clicking.
|
|
cmd /k python.exe kohya_gui.py %*
|
|
) ELSE (
|
|
REM echo This script was started from a command prompt.
|
|
python.exe kohya_gui.py %*
|
|
)
|
|
)
|