[Admin maintenance] Support new ZeroGPU hardware

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