feat(policies): add autoregressive VLAs with tokenization PiFast (#2734)

This commit is contained in:
Jade Choghari
2026-01-09 23:08:37 +01:00
committed by GitHub
parent ba3d2148a3
commit 1d86c9b7f2
15 changed files with 3214 additions and 5 deletions

View File

@@ -75,7 +75,7 @@ from .policy_robot_bridge import (
RobotActionToPolicyActionProcessorStep,
)
from .rename_processor import RenameObservationsProcessorStep
from .tokenizer_processor import TokenizerProcessorStep
from .tokenizer_processor import ActionTokenizerProcessorStep, TokenizerProcessorStep
__all__ = [
"ActionProcessorStep",
@@ -122,6 +122,7 @@ __all__ = [
"AddBatchDimensionProcessorStep",
"RobotProcessorPipeline",
"TokenizerProcessorStep",
"ActionTokenizerProcessorStep",
"Torch2NumpyActionProcessorStep",
"RobotActionToPolicyActionProcessorStep",
"PolicyActionToRobotActionProcessorStep",