diff --git a/src/openpi/training/config.py b/src/openpi/training/config.py index a59cd1cf67..a691332f10 100644 --- a/src/openpi/training/config.py +++ b/src/openpi/training/config.py @@ -702,7 +702,7 @@ def __post_init__(self) -> None: model=pi0_fast.Pi0FASTConfig(action_dim=7, action_horizon=10, max_token_len=180), data=LeRobotLiberoDataConfig( repo_id="physical-intelligence/libero", - base_config=DataConfig(prompt_from_task=True), + base_config=DataConfig(prompt_from_task=True, use_quantile_norm=True), extra_delta_transform=True, ), # Note that we load the pi0-FAST base model checkpoint here. @@ -718,7 +718,7 @@ def __post_init__(self) -> None: ), data=LeRobotLiberoDataConfig( repo_id="physical-intelligence/libero", - base_config=DataConfig(prompt_from_task=True), + base_config=DataConfig(prompt_from_task=True, use_quantile_norm=True), extra_delta_transform=True, ), weight_loader=weight_loaders.CheckpointWeightLoader("gs://openpi-assets/checkpoints/pi0_fast_base/params"),