meet error 'PI0Config' object has no attribute 'normalization_mapping'

[THTHDGCS]

When I run demo command provided in "Minimal Validation Training":
torchrun --nnodes 1 --nproc_per_node 1 \ # number of processes per node, e.g., 1 scripts/train/train.py \ --config run_configs/train/pi0_genmanip_v1.yaml # Config file that specifies which model to train on which dataset, along with hyperparameters

I meet this error: "AttributeError: 'PI0Config' object has no attribute 'normalization_mapping". And this error occurs when running model = AutoModel.from_pretrained(config.base_model_path, **kwargs) , I would like to know how to solve it.

[JiantongChen]

We are reviewing the code. This issue will be documented and resolved as soon as possible.