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_models.py

@@ -34,11 +34,18 @@ class ModelTest(unittest.TestCase):
         model = self.models.get_model("LARGE_1_1")
 
         self.assertEqual(model.license_name, "CC BY-NC 4.0")
+        self.assertFalse(model.enabled)
 
     def test_giant_models_expose_gaussian_head(self):
         for model_key in ("GIANT_1_1", "NESTED_GIANT_LARGE_1_1"):
             with self.subTest(model_key=model_key):
                 self.assertTrue(self.models.get_model(model_key).supports_gs)
+                self.assertFalse(self.models.get_model(model_key).enabled)
+
+    def test_only_uploaded_da3_models_are_enabled(self):
+        enabled = [key for key, model in self.models.MODELS.items() if model.enabled]
+
+        self.assertEqual(enabled, ["SMALL", "BASE"])
 
     def test_unknown_model_is_rejected(self):
         with self.assertRaisesRegex(ValueError, "Unknown DA3 model"):
@@ -72,3 +79,22 @@ class ModelTest(unittest.TestCase):
             self.models.model_directory_name("BEN2_BASE"),
             "ben2_base",
         )
+
+    def test_moge2_uses_single_checkpoint_and_mit_license(self):
+        model = self.models.get_downloadable_model("MOGE2_VITB_NORMAL")
+
+        self.assertEqual(model.repository, "Ruicheng/moge-2-vitb-normal")
+        self.assertEqual(model.license_name, "MIT")
+        self.assertEqual(model.download_patterns, ("model.pt",))
+        self.assertEqual(model.ready_patterns, ("model.pt",))
+        self.assertFalse(model.supports_gs)
+
+    def test_moge2_models_match_uploaded_r2_variants(self):
+        self.assertEqual(
+            list(self.models.MOGE2_MODELS),
+            ["MOGE2_VITS_NORMAL", "MOGE2_VITB_NORMAL", "MOGE2_VITL_NORMAL"],
+        )
+        self.assertEqual(
+            [model.r2_directory for model in self.models.MOGE2_MODELS.values()],
+            ["MoGe2ViTSNormal", "MoGe2ViTBNormal", "MoGe2ViTLNormal"],
+        )