Use UV instead of pip for invokeai

H3mul · Oct 21, 2024 · 1bfb7dd · 1bfb7dd
1 parent 624c2b8
commit 1bfb7dd
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -78,6 +78,9 @@ RUN wget https://github.com/runpod/runpodctl/releases/download/v1.10.0/runpodctl
 RUN curl -s https://packagecloud.io/install/repositories/ookla/speedtest-cli/script.deb.sh | bash && \
     apt install speedtest
 
+
+RUN pip install uv
+
 # Remove existing SSH host keys
 RUN rm -f /etc/ssh/ssh_host_*
 RUN mkdir -p /app/{pid,config,scripts}

diff --git a/scripts/provision_invokeai.sh b/scripts/provision_invokeai.sh
@@ -17,10 +17,10 @@ fi
 
 source venv/bin/activate
 
-pip3 install "InvokeAI[xformers]==${INVOKEAI_PIPY_VERSION}" --use-pep517
+uv pip install "InvokeAI[xformers]==${INVOKEAI_PIPY_VERSION}"
 
 [ -f invokeai.yaml ] || cp /app/config/invokeai/invokeai.yaml ./invokeai.yaml
 
-pip3 cache purge
+uv pip cache purge
 deactivate
 touch install_complete