Prevent VAE inference when running in offline mode. In the lerobot dataset, the presence of the 'action' field incorrectly triggers the VAE inference block. This leads to a RuntimeError due to mismatched tensor dimensions (3 vs 2) when concatenating cls_embed, robot_state_embed, and action_embed—since action_embed lacks the chunk_size dimension. Additionally, this aligns with the original paper, where variational inference is skipped during inference.
* fix: decorator calls with parentheses
* fix no grad for normalize too
Signed-off-by: Francesco Capuano <74058581+fracapuano@users.noreply.github.com>
---------
Signed-off-by: Francesco Capuano <74058581+fracapuano@users.noreply.github.com>
