Upload Dockerfile

Dockerfile

@@ -56,6 +56,8 @@ RUN pyenv install $PYTHON_VERSION && \
 	pip install insightface && \
 	pip install deepdiff && \
 	pip install addict && \
+	pip install pynvml && \
+	pip install yapf && \
 	pip install timm
 	#pip install huggingface_hub["cli"] && \
 	#huggingface-cli delete-cache
@@ -197,7 +199,9 @@ RUN echo "Downloading checkpoints..." && \
 		git clone https://github.com/Nourepide/ComfyUI-Allor.git && \
 		git clone https://github.com/storyicon/comfyui_segment_anything.git && \
 	    git clone https://github.com/crystian/ComfyUI-Crystools && \
-		git clone https://github.com/EllangoK/ComfyUI-post-processing-nodes
+		git clone https://github.com/EllangoK/ComfyUI-post-processing-nodes && \
+		git clone https://github.com/melMass/comfy_mtb.git
+		
     # Install custom nodes
     RUN echo "Installing custom nodes..."