feat: Add MoGe-2 support and enhance depth plane functionality

- Introduced new MoGe-2 depth plane group with geometry and material displacement options.
- Updated existing depth plane group to improve background depth handling and subdivision settings.
- Enhanced tests to validate MoGe-2 models and ensure proper integration with existing systems.
- Modified installation environment to include new dependencies for MoGe-2 and utils3d.
- Updated vendor lock file to include new artifacts for MoGe and utils3d.
- Refactored download and model handling to accommodate new MoGe-2 models.
- Improved validation scripts for depth plane assets to ensure compliance with new specifications.

tests/test_worker.py

@@ -436,7 +436,13 @@ class WorkerTest(unittest.TestCase):
             with self.assertRaisesRegex(RuntimeError, "Unsupported input type"):
                 self.worker.prepare_input(input_path, 16, directory)
 
-    def test_download_model_skips_unreachable_endpoint(self):
+    def test_moge_runtime_disables_unsupported_xformers_attention(self):
+        with patch.dict(os.environ, {}, clear=True):
+            self.worker.prepare_moge_runtime()
+
+            self.assertEqual(os.environ["XFORMERS_DISABLED"], "1")
+
+    def test_download_model_uses_actual_request_without_preflight(self):
         calls = []
         fake_hugging_face = ModuleType("huggingface_hub")
         fake_hugging_face.snapshot_download = lambda **options: calls.append(
@@ -457,13 +463,6 @@ class WorkerTest(unittest.TestCase):
             "create_download_progress_class",
             return_value=object,
         ), patch.dict(os.environ, {"HF_ENDPOINT": ""}), patch.object(
-            self.worker,
-            "hf_endpoint_reachable",
-            return_value=False,
-        ), patch.object(
-            self.worker,
-            "download_from_modelscope",
-        ) as mocked_modelscope, patch.object(
             self.worker,
             "_verify_model_complete",
             return_value=None,
@@ -472,8 +471,7 @@ class WorkerTest(unittest.TestCase):
             args.status = str(Path(directory) / "status.json")
             self.worker.download_model(args)
 
-        self.assertEqual(calls, [])
-        self.assertEqual(mocked_modelscope.call_count, 1)
+        self.assertEqual(len(calls), 1)
 
     def test_download_model_refuses_success_when_files_missing(self):
         fake_hugging_face = ModuleType("huggingface_hub")
@@ -492,10 +490,7 @@ class WorkerTest(unittest.TestCase):
             self.worker,
             "create_download_progress_class",
             return_value=object,
-        ), patch.dict(os.environ, {"HF_ENDPOINT": ""}), patch.object(
-            self.worker,
-            "download_from_modelscope",
-        ):
+        ), patch.dict(os.environ, {"HF_ENDPOINT": ""}):
             args.model_dir = str(Path(directory) / "model")
             args.status = str(Path(directory) / "status.json")
             with self.assertRaisesRegex(
@@ -527,39 +522,35 @@ class WorkerTest(unittest.TestCase):
             return_value=object,
         ), patch.dict(os.environ, {"HF_ENDPOINT": ""}), patch.object(
             self.worker,
-            "download_from_modelscope",
+            "download_from_r2",
             side_effect=OSError("mirror down"),
         ):
+            args.r2_directory = "DA3Small"
+            args.r2_file = ["config.json|100|abc"]
             args.model_dir = str(Path(directory) / "model")
             args.status = str(Path(directory) / "status.json")
-            with self.assertRaisesRegex(RuntimeError, "modelscope.cn"):
+            with self.assertRaisesRegex(RuntimeError, "models.omoolab.xyz"):
                 self.worker.download_model(args)
 
-    def test_download_from_modelscope_downloads_selected_files(self):
+    def test_download_from_r2_downloads_declared_files(self):
         args = SimpleNamespace(
             repository="depth-anything/DA3-SMALL",
             model_dir="model",
             status="status.json",
             allow_pattern=["config.json"],
+            r2_directory="DA3Small",
+            r2_file=["config.json|100|abc"],
         )
         downloaded = []
 
         with tempfile.TemporaryDirectory() as directory, patch.object(
             self.worker,
-            "modelscope_file_list",
-            return_value=[
-                ("config.json", 100),
-                ("model.safetensors", 200),
-                (".gitattributes", 10),
-            ],
-        ), patch.object(
-            self.worker,
-            "_download_modelscope_file",
+            "_download_r2_file",
             side_effect=lambda *args, **kwargs: downloaded.append(args[1]),
         ), patch.object(self.worker, "write_status"):
             args.model_dir = str(Path(directory) / "model")
             args.status = str(Path(directory) / "status.json")
-            self.worker.download_from_modelscope(args, args.model_dir)
+            self.worker.download_from_r2(args, args.model_dir)
 
         self.assertEqual(downloaded, ["config.json"])