golden/trace.json

{
  "vae": {
    "out": [
      [
        1,
        48,
        1,
        14,
        28
      ],
      "torch.float32"
    ]
  },
  "video_pre": {
    "out": "<class 'dict'>"
  },
  "video_prefill": {
    "out": "<class 'list'>"
  },
  "action_step": {
    "out": [
      [
        1,
        32,
        7
      ],
      "torch.float32"
    ]
  }
}