Add Mega-Sam image

.github/workflows/cw-mega-sam.yml

@@ -0,0 +1,25 @@
+on:
+  workflow_dispatch:
+    inputs:
+      base-image:
+        description: "Base image to use"
+        required: true
+      commit:
+        description: "Commit of Mega-sam to include"
+        required: true
+  push:
+    paths:
+      - "cw-mega-sam/**"
+      - ".github/workflows/cw-mega-sam.yml"
+      - ".github/workflows/build.yml"
+
+jobs:
+  build:
+    uses: ./.github/workflows/build.yml
+    secrets: inherit
+    with:
+      image-name: cw-mega-sam
+      folder: cw-mega-sam
+      build-args: |
+        BASE_IMAGE=${{ inputs.base-image || 'ghcr.io/coreweave/ml-containers/torch-extras:es-actions-68fbfd1-nccl-cuda12.4.1-ubuntu22.04-nccl2.25.1-1-torch2.6.0-vision0.21.0-audio2.6.0-abi0'}}
+        COMMIT=${{ inputs.commit || 'main'}}

cw-mega-sam/Dockerfile

@@ -0,0 +1,23 @@
+ARG BASE_IMAGE
+FROM $BASE_IMAGE
+
+RUN mkdir -p /work
+COPY ./cuda124.patch /work/cuda124.patch
+COPY ./requirements.txt /work/requirements.txt
+
+RUN pip install -r /work/requirements.txt
+RUN rm /work/requirements.txt
+
+ARG COMMIT
+RUN cd /work && git clone --recursive https://github.com/mega-sam/mega-sam && \
+    cd mega-sam && \
+    git checkout ${COMMIT}
+RUN cd /work/mega-sam && ls -la
+RUN cd /work/mega-sam && patch -p2 < /work/cuda124.patch
+
+
+
+ENTRYPOINT /work
+CMD echo "Hello! You should only need to run python setup.py install from the /work/mega-sam/base directory."
+
+