admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-23 09:46:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove molmoact2 processor override from factory

Browse Source
This commit is contained in:
hq-fang
2026-05-19 21:58:22 +00:00
parent c092194cf2
commit 140cf2a420
1 changed files with 0 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/policies/factory.py
-19
View File
@@ -294,25 +294,6 @@ def make_pre_post_processors(
kwargs["preprocessor_overrides"] = preprocessor_overrides kwargs["preprocessor_overrides"] = preprocessor_overrides
kwargs["postprocessor_overrides"] = postprocessor_overrides kwargs["postprocessor_overrides"] = postprocessor_overrides
if isinstance(policy_cfg, MolmoAct2Config):
from .molmoact2 import processor_molmoact2 # noqa: F401
preprocessor_overrides = dict(kwargs.get("preprocessor_overrides", {}))
if "normalizer_processor" in preprocessor_overrides:
preprocessor_overrides.setdefault(
"molmoact2_masked_normalizer",
preprocessor_overrides.pop("normalizer_processor"),
)
kwargs["preprocessor_overrides"] = preprocessor_overrides
postprocessor_overrides = dict(kwargs.get("postprocessor_overrides", {}))
if "unnormalizer_processor" in postprocessor_overrides:
postprocessor_overrides.setdefault(
"molmoact2_masked_unnormalizer",
postprocessor_overrides.pop("unnormalizer_processor"),
)
kwargs["postprocessor_overrides"] = postprocessor_overrides
preprocessor = PolicyProcessorPipeline.from_pretrained( preprocessor = PolicyProcessorPipeline.from_pretrained(
pretrained_model_name_or_path=pretrained_path, pretrained_model_name_or_path=pretrained_path,
config_filename=kwargs.get( config_filename=kwargs.get(