Complete wheel-based deployment: add pre-built fused-ssim wheel

- Add pre-built fused_ssim wheel to avoid compilation in Docker
- Update pyproject.docker.toml to use both gsplat and fused_ssim wheels
- Download both wheels in Dockerfile for zero-compilation build
- Regenerate uv.docker.lock with both wheel dependencies
- This eliminates all CUDA compilation during Docker build (fixes exit code 137)

Dockerfile

@@ -46,10 +46,12 @@ ENV HOME=/home/user \
 COPY --chown=user pyproject.docker.toml ./pyproject.toml
 COPY --chown=user uv.docker.lock ./uv.lock
 
-# Download optimized gsplat wheel (built with same PyTorch version)
+# Download pre-built wheels (built with same PyTorch version)
 RUN mkdir -p wheels && \
     wget -O wheels/gsplat-0.1.0-cp310-cp310-linux_x86_64.whl \
-    "https://huggingface.co/blanchon/image-gs-models-utils/resolve/main/gsplat-0.1.0-cp310-cp310-linux_x86_64.whl"
+    "https://huggingface.co/blanchon/image-gs-models-utils/resolve/main/gsplat-0.1.0-cp310-cp310-linux_x86_64.whl" && \
+    wget -O wheels/fused_ssim-0.0.0-cp310-cp310-linux_x86_64.whl \
+    "https://huggingface.co/blanchon/image-gs-models-utils/resolve/main/fused_ssim-0.0.0-cp310-cp310-linux_x86_64.whl"
 
 # Install dependencies with exact versions from lock file
 RUN --mount=type=cache,target=/tmp/uv-cache,sharing=locked,uid=1000,gid=1000 \

pyproject.docker.toml

@@ -23,7 +23,7 @@ dependencies = [
 
 # We use python 3.10 and cu124
 [tool.uv.sources]
-fused_ssim = { git = "https://github.com/rahul-goel/fused-ssim/" }
+fused_ssim = { path = "wheels/fused_ssim-0.0.0-cp310-cp310-linux_x86_64.whl" }
 gsplat = { path = "wheels/gsplat-0.1.0-cp310-cp310-linux_x86_64.whl" }
 torch = [
     { index = "pytorch-cu124", marker = "sys_platform == 'linux'" },

uv.docker.lock

@@ -548,7 +548,10 @@ wheels = [
 [[package]]
 name = "fused-ssim"
 version = "0.0.0"
-source = { git = "https://github.com/rahul-goel/fused-ssim/#b4fd8324e81c48c9b2b9f62e1b9c6431fece6ab3" }
+source = { path = "wheels/fused_ssim-0.0.0-cp310-cp310-linux_x86_64.whl" }
+wheels = [
+    { filename = "fused_ssim-0.0.0-cp310-cp310-linux_x86_64.whl", hash = "sha256:d8330ff7feeba7a3f9ca662dac41f16aa23ce5350c370407842857da21671b87" },
+]
 
 [[package]]
 name = "gradio"
@@ -759,7 +762,7 @@ dev = [
 
 [package.metadata]
 requires-dist = [
-    { name = "fused-ssim", git = "https://github.com/rahul-goel/fused-ssim/" },
+    { name = "fused-ssim", path = "wheels/fused_ssim-0.0.0-cp310-cp310-linux_x86_64.whl" },
     { name = "gradio", specifier = ">=4.0.0" },
     { name = "gsplat", path = "wheels/gsplat-0.1.0-cp310-cp310-linux_x86_64.whl" },
     { name = "huggingface-hub", specifier = ">=0.24.0" },