Skip to content

Commit

Permalink
Replace all pip3 usages to uv
Browse files Browse the repository at this point in the history
  • Loading branch information
H3mul committed Oct 21, 2024
1 parent 1bfb7dd commit 625c055
Show file tree
Hide file tree
Showing 4 changed files with 21 additions and 21 deletions.
20 changes: 10 additions & 10 deletions scripts/provision_a1111.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -40,30 +40,30 @@ 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

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
8 changes: 4 additions & 4 deletions scripts/provision_comfyui.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -28,14 +28,14 @@ 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

# Link output to input for convenience
rm -r input
ln -s output input

pip3 cache purge
uv cache clean
deactivate
touch install_complete
4 changes: 2 additions & 2 deletions scripts/provision_invokeai.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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
10 changes: 5 additions & 5 deletions scripts/provision_kohya.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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

0 comments on commit 625c055

Please sign in to comment.