mirror of
https://github.com/MackinationsAi/UDAV2-ControlNet.git
synced 2026-01-26 23:39:45 +00:00
40 lines
1.9 KiB
Batchfile
40 lines
1.9 KiB
Batchfile
@echo off
|
|
REM Check if the virtual environment folder exists
|
|
IF NOT EXIST venv (
|
|
echo Creating virtual environment...
|
|
python -m venv venv
|
|
)
|
|
|
|
REM Activate the virtual environment
|
|
echo Activating virtual environment...
|
|
CALL venv\Scripts\activate
|
|
|
|
REM Upgrade pip to the latest version
|
|
python.exe -m pip install --upgrade pip --no-cache-dir
|
|
|
|
REM Create checkpoints directory if it does not exist
|
|
IF NOT EXIST checkpoints (
|
|
mkdir checkpoints
|
|
)
|
|
|
|
REM Download models and place them into the checkpoints folder & Download Triton wheel and place it in the main tree folder
|
|
echo Downloading Depth-Anything-V2-Small model...
|
|
curl -L -o checkpoints/depth_anything_v2_vits.safetensors https://huggingface.co/MackinationsAi/Depth-Anything-V2_Safetensors/resolve/main/depth_anything_v2_vits.safetensors?download=true
|
|
echo Downloading Depth-Anything-V2-Base model...
|
|
curl -L -o checkpoints/depth_anything_v2_vitb.safetensors https://huggingface.co/MackinationsAi/Depth-Anything-V2_Safetensors/resolve/main/depth_anything_v2_vitb.safetensors?download=true
|
|
echo Downloading Depth-Anything-V2-Large model...
|
|
curl -L -o checkpoints/depth_anything_v2_vitl.safetensors https://huggingface.co/MackinationsAi/Depth-Anything-V2_Safetensors/resolve/main/depth_anything_v2_vitl.safetensors?download=true
|
|
echo Downloading Triton==2.1.0 .whl...
|
|
curl -L -o triton-2.1.0-cp310-cp310-win_amd64.whl https://huggingface.co/MonsterMMORPG/SECourses/resolve/main/triton-2.1.0-cp310-cp310-win_amd64.whl?download=true
|
|
|
|
REM Installing torch, dependencies from requirements.txt & triton
|
|
echo Installing dependencies...
|
|
pip install torch==2.3.0+cu121 torchvision==0.18.0+cu121 xformers==0.0.26.post1 --index-url https://download.pytorch.org/whl/cu121 --no-cache-dir
|
|
pip install -r requirements.txt --no-cache-dir
|
|
pip install triton-2.1.0-cp310-cp310-win_amd64.whl --no-cache-dir
|
|
|
|
REM Deactivate the virtual environment
|
|
CALL venv\scripts\deactivate
|
|
pause
|
|
exit
|