Upload Dockerfile

Dockerfile

@@ -50,7 +50,9 @@ RUN pyenv install $PYTHON_VERSION && \
 	pip install torchsde && \
 	pip install ultralytics && \
 	pip install simpleeval && \
-	pip install gray2color
+	pip install gray2color && \
+	pip install segment-anything && \
+	pip install typing-aliases
 	#pip install huggingface_hub["cli"] && \
 	#huggingface-cli delete-cache
 	
@@ -190,7 +192,6 @@ RUN echo "Downloading checkpoints..." && \
 		git clone https://github.com/Gourieff/comfyui-reactor-node.git && \
 		git clone https://github.com/Nourepide/ComfyUI-Allor.git && \
 		git clone https://github.com/storyicon/comfyui_segment_anything.git
-		# git clone https://github.com/Niutonian/ComfyUi-NoodleWebcam.git
 	
     # Install custom nodes
     RUN echo "Installing custom nodes..."
@@ -224,6 +225,7 @@ RUN echo "Downloading checkpoints..." && \
 	# RUN cd custom_nodes && wget -c https://huggingface.co/guoyww/animatediff/resolve/main/v2_lora_ZoomOut.ckpt -P ./custom_nodes/ComfyUI-AnimateDiff-Evolved/motion_lora/
   
     RUN echo "Done"
+	RUN pip install torch==2.0.0+cu118 torchvision==0.15.1+cu118 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118
 
 CMD ["python", "main.py", "--cpu", "--listen", "0.0.0.0", "--cpu", "--port", "7860", "--use-split-cross-attention", "--output-directory", "${USE_PERSISTENT_DATA:+/data/}"]