Update Dockerfile

Dockerfile

@@ -29,7 +29,8 @@ RUN pip install torch==2.5.0+cu124 torchvision==0.20.0+cu124 torchaudio==2.5.0+c
     pip install opencv-python imageio imageio-ffmpeg ffmpeg-python av runpod && \
     pip install torchsde einops diffusers transformers accelerate && \
     git clone https://github.com/comfyanonymous/ComfyUI /content/ComfyUI && \
-    git clone https://github.com/kijai/ComfyUI-MochiWrapper /content/ComfyUI/custom_nodes/ComfyUI-MochiWrapper && \
+    # git clone https://github.com/kijai/ComfyUI-MochiWrapper /content/ComfyUI/custom_nodes/ComfyUI-MochiWrapper && \
+    git clone -b dev https://github.com/camenduru/ComfyUI-MochiWrapper /content/ComfyUI/custom_nodes/ComfyUI-MochiWrapper && \
     git clone https://github.com/Kosinkadink/ComfyUI-VideoHelperSuite /content/ComfyUI/custom_nodes/ComfyUI-VideoHelperSuite && \
     git clone https://github.com/kijai/ComfyUI-KJNodes /content/ComfyUI/custom_nodes/ComfyUI-KJNodes && \
     git clone https://github.com/ltdrdata/ComfyUI-Manager /content/ComfyUI/custom_nodes/ComfyUI-Manager && \