Test

README.md

@@ -4,11 +4,11 @@ emoji: 📚
 colorFrom: indigo
 colorTo: green
 sdk: gradio
-sdk_version: 5.50.0
+sdk_version: 4.26.0
 app_file: app.py
 pinned: false
 short_description: Create a 3D model from an image in 10 seconds!
 license: apache-2.0
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -13,57 +13,6 @@ from einops import rearrange, repeat
 from tqdm import tqdm
 from diffusers import DiffusionPipeline, EulerAncestralDiscreteScheduler
 
-import subprocess, sys, glob, tempfile, ctypes
-
-CUDA_HOME = "/cuda-image/usr/local/cuda-13.0"
-CUDA_LIBDIR = os.path.join(CUDA_HOME, "lib64")
-
-@spaces.GPU(duration=210)
-def first_gpu_setup():
-    try:
-        import nvdiffrast  # noqa: F401
-        print("nvdiffrast already installed; skipping build.")
-        return
-    except ImportError:
-        pass
-
-    if not os.path.exists(os.path.join(CUDA_HOME, "bin", "nvcc")):
-        raise RuntimeError(
-            f"nvcc not found at {CUDA_HOME}/bin/nvcc on the ZeroGPU worker. "
-            "The new-hardware CUDA path may have moved; please update CUDA_HOME."
-        )
-
-    patch_dir = tempfile.mkdtemp(prefix="torch_cuda_patch_")
-    with open(os.path.join(patch_dir, "sitecustomize.py"), "w") as f:
-        f.write(
-            "try:\n"
-            "    import torch.utils.cpp_extension as _c\n"
-            "    _c._check_cuda_version = lambda *a, **k: None\n"
-            "except Exception:\n"
-            "    pass\n"
-        )
-
-    env = os.environ.copy()
-    env["CUDA_HOME"] = CUDA_HOME
-    env["CUDA_PATH"] = CUDA_HOME
-    env["PATH"] = os.path.join(CUDA_HOME, "bin") + os.pathsep + env.get("PATH", "")
-    env["PYTHONPATH"] = patch_dir + os.pathsep + env.get("PYTHONPATH", "")
-    env["TORCH_CUDA_ARCH_LIST"] = "12.0"
-
-    subprocess.check_call(
-        [sys.executable, "-m", "pip", "install", "--no-deps",
-         "setuptools", "wheel", "ninja"],
-    )
-    subprocess.check_call(
-        [sys.executable, "-m", "pip", "install", "--no-build-isolation",
-         "git+https://github.com/NVlabs/nvdiffrast/"],
-        env=env,
-    )
-first_gpu_setup()
-
-ctypes.CDLL(os.path.join(CUDA_LIBDIR, "libcudart.so.13"), mode=ctypes.RTLD_GLOBAL)
-os.environ["LD_LIBRARY_PATH"] = CUDA_LIBDIR + os.pathsep + os.environ.get("LD_LIBRARY_PATH", "")
-
 from src.utils.train_util import instantiate_from_config
 from src.utils.camera_util import (
     FOV_to_intrinsics, 
@@ -80,6 +29,7 @@ from huggingface_hub import hf_hub_download
 
 import gradio as gr
 
+
 def get_render_cameras(batch_size=1, M=120, radius=2.5, elevation=10.0, is_flexicubes=False):
     """
     Get the rendering camera parameters.

requirements.txt

@@ -1,7 +1,7 @@
-torch==2.8.0
-torchvision==0.23.0
-torchaudio==2.8.0
-pytorch-lightning
+torch==2.1.0
+torchvision==0.16.0
+torchaudio==2.1.0
+pytorch-lightning==2.1.2
 einops
 omegaconf
 deepspeed
@@ -12,11 +12,12 @@ tensorboard
 PyMCubes
 trimesh
 rembg
-transformers==4.46.3
-diffusers==0.32.1
+transformers==4.34.1
+diffusers==0.19.3
 bitsandbytes
 imageio[ffmpeg]
 xatlas
 plyfile
-xformers
-onnxruntime
+xformers==0.0.22.post7
+git+https://github.com/NVlabs/nvdiffrast/
+huggingface-hub