fix(pipeline): do not treat ckpts/... as Hugging Face org/model id

- _is_hub_model_id: exclude ckpts/, weights/, configs/, checkpoints/
- Load models from snapshot root only for relative paths; remove broad try/except fallback
- Fixes 404 on hf.co/ckpts/... when NeAR CPU preload runs on Space

Made-with: Cursor

tests/test_pipeline_hub_id.py

@@ -0,0 +1,36 @@
+"""Regression: pipeline model paths like ckpts/foo must not be treated as Hub org/model."""
+
+from __future__ import annotations
+
+import re
+import unittest
+from pathlib import Path
+
+
+_BASE_PATH = Path(__file__).resolve().parents[1] / "trellis/pipelines/base.py"
+
+
+class PipelineHubIdTests(unittest.TestCase):
+    def test_base_py_documents_ckpts_exclusion(self) -> None:
+        src = _BASE_PATH.read_text(encoding="utf-8")
+        self.assertIn("_NOT_HUB_PATH_PREFIXES", src)
+        self.assertIn('"ckpts/"', src)
+        self.assertIn("models.from_pretrained(os.path.join(root, vp))", src)
+
+    def test_regex_ckpts_two_segment_not_hub_semantics(self) -> None:
+        """Mirror _is_hub_model_id: two-segment ckpts/... is not a Hub id."""
+        not_hub = ("ckpts/", "weights/", "configs/", "checkpoints/")
+
+        def is_hub(t: str) -> bool:
+            t = t.strip().replace("\\", "/")
+            if not re.fullmatch(r"[\w.-]+/[\w.-]+", t):
+                return False
+            tl = t.lower()
+            return not any(tl.startswith(p) for p in not_hub)
+
+        self.assertFalse(is_hub("ckpts/slat_dec_gs_swin8_B_64l8gs32_fp16"))
+        self.assertTrue(is_hub("luh0502/NeAR"))
+
+
+if __name__ == "__main__":
+    unittest.main()

trellis/pipelines/base.py

@@ -10,10 +10,22 @@ import torch.nn as nn
 from .. import models
 
 
+# Paths under a snapshot (e.g. ckpts/foo) also match org/model regex — exclude them.
+_NOT_HUB_PATH_PREFIXES = (
+    "ckpts/",
+    "weights/",
+    "configs/",
+    "checkpoints/",
+)
+
+
 def _is_hub_model_id(s: str) -> bool:
-    """`org/model` style id, not a filesystem path."""
+    """True for Hugging Face ``org/model``, not for relative snapshot paths like ``ckpts/...``."""
     t = s.strip().replace("\\", "/")
-    return bool(re.fullmatch(r"[\w.-]+/[\w.-]+", t))
+    if not re.fullmatch(r"[\w.-]+/[\w.-]+", t):
+        return False
+    tl = t.lower()
+    return not any(tl.startswith(p) for p in _NOT_HUB_PATH_PREFIXES)
 
 
 class Pipeline:
@@ -85,10 +97,11 @@ class Pipeline:
 
         _models = {}
         for k, v in args["models"].items():
-            try:
-                _models[k] = models.from_pretrained(os.path.join(root, v))
-            except Exception:
-                _models[k] = models.from_pretrained(v)
+            vp = str(v).strip().replace("\\", "/")
+            if _is_hub_model_id(vp):
+                _models[k] = models.from_pretrained(vp)
+            else:
+                _models[k] = models.from_pretrained(os.path.join(root, vp))
 
         new_pipeline = Pipeline(_models)
         new_pipeline._pretrained_args = args