diff --git a/scripts/provision_a1111.sh b/scripts/provision_a1111.sh index 89df42a..4c935df 100644 --- a/scripts/provision_a1111.sh +++ b/scripts/provision_a1111.sh @@ -19,9 +19,9 @@ source venv/bin/activate cp /app/config/a1111/{requirements.txt,requirements_versions.txt,install-automatic.py} ./ -pip3 install torch==2.0.1 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 -pip3 install xformers==0.0.22 tensorrt -pip3 install -r requirements_versions.txt +uv pip install torch==2.0.1 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 +uv pip install xformers==0.0.22 tensorrt +uv pip install -r requirements_versions.txt echo "Initial reqs complete, running A1111 install..." python3 -m install-automatic --skip-torch-cuda-test @@ -40,15 +40,15 @@ echo "Installing plugins..." echo "Installing plugin dependencies..." cd ${A1111_ROOT}/extensions/deforum -pip3 install -r requirements.txt +uv pip install -r requirements.txt cd ${A1111_ROOT}/extensions/sd-webui-controlnet -pip3 install -r requirements.txt +uv pip install -r requirements.txt cd ${A1111_ROOT}/extensions/sd-webui-reactor -pip3 install -r requirements.txt -pip3 install onnxruntime-gpu +uv pip install -r requirements.txt +uv pip install onnxruntime-gpu echo "CUDA" > last_device.txt cd ${A1111_ROOT}/extensions/infinite-image-browsing -pip3 install -r requirements.txt +uv pip install -r requirements.txt cd ${A1111_ROOT}/extensions/adetailer python3 -m install @@ -56,14 +56,14 @@ cd ${A1111_ROOT}/extensions/sd_dreambooth_extension git checkout main git reset ${DREAMBOOTH_COMMIT} --hard cp /app/config/a1111/requirements_dreambooth.txt ./requirements.txt -pip3 install -r requirements.txt +uv pip install -r requirements.txt cd ${A1111_ROOT} echo "Finished installing plugins..." cp /app/config/a1111/{webui-user.sh,config.json,ui-config.json} ./ -pip3 cache purge +uv cache clean deactivate touch install_complete diff --git a/scripts/provision_comfyui.sh b/scripts/provision_comfyui.sh index fdb8c3f..b0fd264 100644 --- a/scripts/provision_comfyui.sh +++ b/scripts/provision_comfyui.sh @@ -17,8 +17,8 @@ fi [ -d venv ] || python3 -m venv --system-site-packages venv source venv/bin/activate -pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu121 -pip3 install -r requirements.txt +uv pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu121 +uv pip install -r requirements.txt git -C ./custom_nodes clone --depth 1 https://github.com/ltdrdata/ComfyUI-Manager.git git -C ./custom_nodes clone --depth 1 https://github.com/XLabs-AI/x-flux-comfyui @@ -28,7 +28,7 @@ git -C ./custom_nodes clone --depth 1 https://github.com/rgthree/rgthree-comfy for dir in $PWD/custom_nodes/*/; do if [ -d "$dir" ] && [ -f "$dir/requirements.txt" ]; then echo "Installing module requirements: $dir" - pip3 install -qq -r $dir/requirements.txt + uv pip install -qq -r $dir/requirements.txt fi done @@ -36,6 +36,6 @@ done rm -r input ln -s output input -pip3 cache purge +uv cache clean deactivate touch install_complete diff --git a/scripts/provision_invokeai.sh b/scripts/provision_invokeai.sh index 0b64c59..396981c 100644 --- a/scripts/provision_invokeai.sh +++ b/scripts/provision_invokeai.sh @@ -17,10 +17,10 @@ fi source venv/bin/activate -uv pip install "InvokeAI[xformers]==${INVOKEAI_PIPY_VERSION}" +uv pip install -v "InvokeAI[xformers]==${INVOKEAI_PIPY_VERSION}" [ -f invokeai.yaml ] || cp /app/config/invokeai/invokeai.yaml ./invokeai.yaml -uv pip cache purge +uv cache clean deactivate touch install_complete \ No newline at end of file diff --git a/scripts/provision_kohya.sh b/scripts/provision_kohya.sh index fa55daf..4398a05 100644 --- a/scripts/provision_kohya.sh +++ b/scripts/provision_kohya.sh @@ -19,16 +19,16 @@ cp /app/config/kohya_ss/requirements* ./ [ -d venv ] || python3 -m venv --system-site-packages venv source venv/bin/activate -pip3 install torch==2.0.1 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 -pip3 install xformers==0.0.22 \ +uv pip install torch==2.0.1 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 +uv pip install xformers==0.0.22 \ bitsandbytes==0.41.1 \ tensorboard==2.14.1 \ tensorflow==2.14.0 \ wheel \ scipy \ tensorrt -pip3 install -r requirements.txt -pip3 install . -pip3 cache purge +uv pip install -r requirements.txt +uv pip install . +uv cache clean deactivate touch install_complete