revert to working version for ply to glb converter and set appropriate rast flag

Dockerfile

@@ -1,4 +1,4 @@
-FROM nvcr.io/nvidia/pytorch:23.06-py3
+FROM nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04
 
 ARG DEBIAN_FRONTEND=noninteractive
 
@@ -10,6 +10,9 @@ COPY ./requirements.txt /code/
 
 RUN apt-get update && apt-get install --no-install-recommends -y \
     build-essential \
+    python3.10 \
+    python3.10-dev \
+    python3-pip \
     git \
     ffmpeg \
     # Install nvdiffrast dependencies

app.py

@@ -17,7 +17,7 @@ import gradio as gr
 from huggingface_hub import hf_hub_download
 ckpt_path = hf_hub_download(repo_id="ashawkey/LGM", filename="model_fp16.safetensors")
 
-subprocess.run(shlex.split("pip install diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl"))
+subprocess.run(shlex.split("pip install wheel/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl"))
 
 import kiui
 from kiui.op import recenter
@@ -196,6 +196,7 @@ def process(input_image, prompt, prompt_neg='', input_elevation=0, input_num_ste
 
     # load a saved ply and convert to mesh
     opt.test_path = output_ply_path
+    opt.force_cuda_rast = True # container compatability
     
     converter = Converter(opt).cuda()
     converter.fit_nerf()

diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl → wheel/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c1c30cbf1d8de77ee8be049d8f8aeb36ccadffc408c080ad79e8341125e3585f
-size 3108178
+oid sha256:42bf718442ba764469170abc09d99a70b7c1d891dc290f2e1247db09c95a0e88
+size 3021758