Enable FCGF with torch 2.11 CUDA 13 MinkowskiEngine

app.py

@@ -33,7 +33,7 @@ PIPELINE_CONFIG_ROOT = Path("/tmp/canis_pipeline_config")
 QUOTA_ROOT = Path(os.environ.get("CANIS_QUOTA_ROOT", "/data/canis-usage"))
 DAILY_CANONICALIZE_LIMIT = int(os.environ.get("CANIS_DAILY_LIMIT", "5"))
 QUEUE_MAX_SIZE = int(os.environ.get("CANIS_QUEUE_MAX_SIZE", "3"))
-DIAGNOSTIC_MODE = "production_fpfh"
+DIAGNOSTIC_MODE = "production_fcgf"
 IS_ZERO_GPU = os.environ.get("SPACES_ZERO_GPU") is not None
 
 _quota_lock = threading.RLock()
@@ -84,7 +84,7 @@ def _canonicalize_on_gpu(
         args = canis_job_entry._ARGS
         if app is None or args is None:
             raise RuntimeError("Private CANIS job context is not initialized")
-        runtime = app.load_runtime(args, need_fcgf=False)
+        runtime = app.load_runtime(args, need_fcgf=True)
         move_started = time.perf_counter()
         _trace("phase=runtime.trellis.move_cpu_to_cuda event=START")
         runtime.ctx.pipeline.to(torch.device("cuda:0"))
@@ -763,7 +763,7 @@ def main() -> None:
     if DIAGNOSTIC_MODE not in {"zero_pack_deferred_native_probe", "zero_pack_pure_python_probe"}:
         _install_pytorch3d_compatibility()
         _install_blackwell_attention_compatibility()
-    _install_fpfh_only_compatibility()
+    _trace("phase=runtime.registration_mode event=READY mode=fcgf minkowski=enabled")
     private_checkpoint = _validate_models()
     checkpoint = _prepare_cached_pipeline_config(private_checkpoint)
     _trace("phase=runtime.model_source event=READY source=public_trellis_hybrid")
@@ -1005,9 +1005,9 @@ def main() -> None:
         "--server-name", "0.0.0.0",
         "--server-port", os.environ.get("PORT", "7860"),
         "--glb-limit", "6",
-        "--registration-desc", "fpfh",
+        "--registration-desc", "fcgf",
     ]
-    if not IS_ZERO_GPU and DIAGNOSTIC_MODE == "production_fpfh":
+    if not IS_ZERO_GPU and DIAGNOSTIC_MODE == "production_fcgf":
         argv.append("--preload")
     os.chdir(RUNTIME_ROOT)
     sys.argv = argv[1:]

requirements.txt

@@ -1,30 +1,31 @@
-torchvision==0.26.0
-numpy==1.26.4
-gradio==5.50.0
-transformers==4.46.3
-pillow==10.4.0
-imageio==2.36.1
-imageio-ffmpeg==0.5.1
-tqdm==4.67.1
-easydict==1.13
-einops==0.8.1
-opencv-python-headless==4.10.0.84
-scipy==1.14.1
-rembg==2.0.60
-onnxruntime==1.20.1
-trimesh==4.5.3
-open3d==0.19.0
-xatlas==0.0.9
-pyvista==0.44.2
-pymeshfix==0.17.0
-igraph==0.11.8
-spconv-cu120==2.3.6
-plotly==5.24.1
-matplotlib==3.9.2
-plyfile==1.1
-safetensors==0.4.5
-ninja==1.11.1.1
-git+https://github.com/EasternJournalist/utils3d.git@9a4eb15e4021b67b12c460c7057d642626897ec8
-https://huggingface.co/Kendong/CANIS-build-deps/resolve/a22329f2bc093a7576f1f7c418290eaeb8383284/torch211-cu130-sm89/pytorch3d-0.7.9-cp310-cp310-linux_x86_64.whl#sha256=c0f1d01381ebd01f29a3b4924a660fd73b31fae606fadfa503d47d5318ee1789
-https://huggingface.co/Kendong/CANIS-build-deps/resolve/68041ac4fb3ff1504baab7f3523eeaf91d7772fa/torch211-cu130-sm89/vox2seq-0.0.0-cp310-cp310-linux_x86_64.whl#sha256=2ef5c6e7dade24573e4e51d93ce28268a0df55128cb9424f9a921c3eccaa6eaa
-https://huggingface.co/Kendong/CANIS-build-deps/resolve/f14cde37840a300b225ed239d791f228dfb001bc/torch211-cu130-sm120/xformers-0.0.35-py39-none-linux_x86_64.whl#sha256=4737772e0d6cf5ee709840dee30a3dc247501181e2725daa4cffd7dee76ce9f9
+torchvision==0.26.0
+numpy==1.26.4
+gradio==5.50.0
+transformers==4.46.3
+pillow==10.4.0
+imageio==2.36.1
+imageio-ffmpeg==0.5.1
+tqdm==4.67.1
+easydict==1.13
+einops==0.8.1
+opencv-python-headless==4.10.0.84
+scipy==1.14.1
+rembg==2.0.60
+onnxruntime==1.20.1
+trimesh==4.5.3
+open3d==0.19.0
+xatlas==0.0.9
+pyvista==0.44.2
+pymeshfix==0.17.0
+igraph==0.11.8
+spconv-cu120==2.3.6
+plotly==5.24.1
+matplotlib==3.9.2
+plyfile==1.1
+safetensors==0.4.5
+ninja==1.11.1.1
+git+https://github.com/EasternJournalist/utils3d.git@9a4eb15e4021b67b12c460c7057d642626897ec8
+https://huggingface.co/Kendong/CANIS-build-deps/resolve/56d6c062fa9d439ec0a39470d600e94bce44e5c9/torch211-cu130-sm89/minkowskiengine-0.5.4-cp310-cp310-linux_x86_64.whl#sha256=b5dae5e0323975fbb261e6ae789a7e6773063051db333fdc1bec0ee7d5456939
+https://huggingface.co/Kendong/CANIS-build-deps/resolve/a22329f2bc093a7576f1f7c418290eaeb8383284/torch211-cu130-sm89/pytorch3d-0.7.9-cp310-cp310-linux_x86_64.whl#sha256=c0f1d01381ebd01f29a3b4924a660fd73b31fae606fadfa503d47d5318ee1789
+https://huggingface.co/Kendong/CANIS-build-deps/resolve/68041ac4fb3ff1504baab7f3523eeaf91d7772fa/torch211-cu130-sm89/vox2seq-0.0.0-cp310-cp310-linux_x86_64.whl#sha256=2ef5c6e7dade24573e4e51d93ce28268a0df55128cb9424f9a921c3eccaa6eaa
+https://huggingface.co/Kendong/CANIS-build-deps/resolve/f14cde37840a300b225ed239d791f228dfb001bc/torch211-cu130-sm120/xformers-0.0.35-py39-none-linux_x86_64.whl#sha256=4737772e0d6cf5ee709840dee30a3dc247501181e2725daa4cffd7dee76ce9f9