update

embodied_gen/utils/monkey_patch/gradio.py

@@ -44,17 +44,29 @@ def _patch_gradio_schema_bool_bug() -> None:
     )
 
 
+# def _patch_open3d_cuda_device_count_bug() -> None:
+#     """Patch open3d to avoid CUDA probing during import."""
+#     with fileinput.FileInput(
+#         f'{site.getsitepackages()[0]}/open3d/__init__.py', inplace=True
+#     ) as file:
+#         for line in file:
+#             print(
+#                 line.replace(
+#                     '_pybind_cuda.open3d_core_cuda_device_count() > 0',
+#                     'False',
+#                 ).replace('if 1 > 0:', 'if False:'),
+#                 end='',
+#             )
+
 def _patch_open3d_cuda_device_count_bug() -> None:
-    """Patch open3d to avoid CUDA probing during import."""
+    """Patch open3d to avoid cuda device count bug."""
     with fileinput.FileInput(
         f'{site.getsitepackages()[0]}/open3d/__init__.py', inplace=True
     ) as file:
         for line in file:
             print(
                 line.replace(
-                    '_pybind_cuda.open3d_core_cuda_device_count() > 0',
-                    'False',
-                ).replace('if 1 > 0:', 'if False:'),
+                    '_pybind_cuda.open3d_core_cuda_device_count()', '1'
+                ),
                 end='',
             )
-

requirements.txt

@@ -61,7 +61,7 @@ seaborn
 hydra-core
 modelscope
 timm
-open3d==0.18.0
+open3d
 MoGe@git+https://github.com/microsoft/MoGe.git@a8c3734
 
 https://huggingface.co/xinjjj/RoboAssetGen/resolve/main/wheel_cu126/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl