暂存

2026-02-09 14:41:35 +08:00
parent f833c6d9f1
commit 8b700b6d99
10 changed files with 76 additions and 39 deletions
--- a/roboimi/vla/models/heads/init.py
+++ b/roboimi/vla/models/heads/init.py
@@ -1,4 +1,4 @@
 # # Action Head models
-from .diffusion import ConditionalUnet1D
+from .conditional_unet1d import ConditionalUnet1D

 __all__ = ["ConditionalUnet1D"]