[Admin maintenance] Support new ZeroGPU hardware

Thank you so much for having shared this Space with the community on this demo. We have upgraded the ZeroGPU infra-structure to run on modern blackwell architecture.
For that, we need to upgrade your demo to support that. This PR fixes your demo to work with the new architecture. As this is something we broke on our end, we may merge this PR autonomously. If this breaks unexpectedly or brings unintended consequences, feel free to revert, modify or otherwise. Any issues you can email apolinario@huggingface.co

app.py

@@ -15,13 +15,12 @@
 
 import os
 os.system("rm -rf /data-nvme/zerogpu-offload/")
-os.system("pip install chumpy")
+os.system("pip install --no-build-isolation chumpy")
 os.system("pip uninstall -y basicsr")
 os.system("pip install git+https://github.com/hitsz-zuoqi/BasicSR/")
 os.system("pip install numpy==1.23.0")
-os.system("pip install ./wheels_new/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl --force-reinstall")
-os.system("pip install ./wheels/simple_knn-0.0.0-cp310-cp310-linux_x86_64.whl --force-reinstall")
-os.system("pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py310_cu121_pyt240/download.html")
+# diff_gaussian_rasterization / pytorch3d / simple_knn are now installed at build time from
+# requirements.txt (prebuilt Blackwell sm_120 wheels); no runtime pip install needed.
 import cv2
 import time
 from PIL import Image
@@ -31,6 +30,14 @@ import base64
 import spaces
 import torch
 torch._dynamo.config.disable = True
+
+# torch>=2.6 flipped torch.load's default to weights_only=True, which breaks loading
+# these (trusted, first-party) checkpoints that pickle argparse.Namespace and other globals.
+_orig_torch_load = torch.load
+def _torch_load_full(*args, **kwargs):
+    kwargs.setdefault("weights_only", False)
+    return _orig_torch_load(*args, **kwargs)
+torch.load = _torch_load_full
 import subprocess
 import os
 import argparse
@@ -400,7 +407,7 @@ def launch_pretrained():
     hf_hub_download(repo_id="3DAIGC/LHM-500M-HF", repo_type='model', filename='model.safetensors', local_dir="./exps/releases/video_human_benchmark/human-lrm-500M/step_060000/")
 
 def launch_env_not_compile_with_cuda():
-    os.system("pip install chumpy")
+    os.system("pip install --no-build-isolation chumpy")
     os.system("pip uninstall -y basicsr")
     os.system("pip install git+https://github.com/hitsz-zuoqi/BasicSR/")
     os.system("pip install numpy==1.23.0")

requirements.txt

@@ -12,7 +12,6 @@ diffusers
 dna==0.0.1
 gfpgan==1.3.8
 gsplat==1.4.0
-huggingface_hub==0.23.2
 imageio==2.19.3
 jaxtyping==0.2.38
 kiui==0.2.14
@@ -38,14 +37,12 @@ setuptools==74.0.0
 taming_transformers_rom1504==0.0.6
 timm==1.0.15
 
-# https://download.pytorch.org/whl/cu121/torch-2.5.1%2Bcu121-cp310-cp310-linux_x86_64.whl#sha256=92af92c569de5da937dd1afb45ecfdd598ec1254cf2e49e3d698cb24d71aae14
-# https://download.pytorch.org/whl/cu121/torchvision-0.20.1%2Bcu121-cp310-cp310-linux_x86_64.whl#sha256=304937b82c933d5155bd04d771f4b187273f67a76050bb4276b521f7e9b4c4e7
-# https://download.pytorch.org/whl/cu121/xformers-0.0.29.post1-cp310-cp310-manylinux_2_28_x86_64.whl#sha256=e213ff8123e20602bd486739ffee4013338b02f9d2e0e4635a2912750854fdbe
-
-https://download.pytorch.org/whl/cu121/torch-2.4.0%2Bcu121-cp310-cp310-linux_x86_64.whl#sha256=28bfba084dca52a06c465d7ad0f3cc372c35fc503f3eab881cc17a5fd82914e7
-https://download.pytorch.org/whl/cu121/torchvision-0.19.0%2Bcu121-cp310-cp310-linux_x86_64.whl#sha256=5ee103c7eb47f8b08837e0e48b178f7ecc91d769d2b61240b90cb5aa2d06ce77
-
-
+# Blackwell (sm_120) ZeroGPU: use stock torch 2.8.0 (default PyPI cu128 wheel supports sm_120)
+# plus prebuilt CUDA-extension wheels from multimodalart/zerogpu-blackwell-wheels (tag pt28-cu128-cp310).
+torch==2.8.0
+torchvision==0.23.0
+https://huggingface.co/datasets/multimodalart/zerogpu-blackwell-wheels/resolve/main/wheels/pt28-cu128-cp310/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl
+https://huggingface.co/datasets/multimodalart/zerogpu-blackwell-wheels/resolve/main/wheels/pt28-cu128-cp310/pytorch3d-0.7.9-cp310-cp310-linux_x86_64.whl
 
 tqdm==4.66.4
 transformers==4.41.2