simplify molmoact2 package imports

This commit is contained in:
hq-fang
2026-05-19 21:57:45 +00:00
parent e870af119f
commit b858ba1b6c
+2 -12
View File
@@ -15,17 +15,7 @@
# limitations under the License.
from .configuration_molmoact2 import MolmoAct2Config
from .modeling_molmoact2 import MolmoAct2Policy
from .processor_molmoact2 import make_molmoact2_pre_post_processors
__all__ = ["MolmoAct2Config", "MolmoAct2Policy", "make_molmoact2_pre_post_processors"]
def __getattr__(name):
if name == "MolmoAct2Policy":
from .modeling_molmoact2 import MolmoAct2Policy
return MolmoAct2Policy
if name == "make_molmoact2_pre_post_processors":
from .processor_molmoact2 import make_molmoact2_pre_post_processors
return make_molmoact2_pre_post_processors
raise AttributeError(name)