chore: install dev container comfystream in edit mode (#68)

yondonfu · Feb 18, 2025 · 0119f59 · 0119f59
1 parent 7ea9d93
commit 0119f59
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/.devcontainer/post-create.sh b/.devcontainer/post-create.sh
@@ -1,7 +1,12 @@
 #!/bin/bash
+
 chmod +x /workspace/comfystream/docker/entrypoint.sh
 cd /workspace/comfystream
 
+# Install Comfystream in editable mode.
+/workspace/comfystream/docker/entrypoint.sh --editable
+
+# Create a symlink to the entrypoint script.
 echo 'alias prepare_examples="/workspace/comfystream/docker/entrypoint.sh --download-models --build-engines"' >> ~/.bashrc
 echo -e "\e[32mContainer ready! Run 'prepare_examples' to download models and build engines for example workflows.\e[0m"
 

diff --git a/docker/entrypoint.sh b/docker/entrypoint.sh
@@ -44,6 +44,13 @@ if [ "$1" = "--build-engines" ]; then
     shift
 fi
 
+if [ "$1" = "--editable" ]; then
+    cd /workspace/comfystream
+    conda activate comfystream
+    pip install -e .
+    shift
+fi
+
 # Install npm packages if needed
 cd /workspace/comfystream/ui
 if [ ! -d "node_modules" ]; then
@@ -52,6 +59,7 @@ fi
 
 if [ "$1" = "--server" ]; then
     /usr/bin/supervisord -c /etc/supervisor/supervisord.conf
+    shift
 fi
 
 cd /workspace/comfystream