Man1103/XVLA-Base-0.88B

1{2  "name": "policy_preprocessor",3  "steps": [4    {5      "registry_name": "rename_observations_processor",6      "config": {7        "rename_map": {}8      }9    },10    {11      "registry_name": "to_batch_processor",12      "config": {}13    },14    {15      "registry_name": "tokenizer_processor",16      "config": {17        "max_length": 50,18        "task_key": "task",19        "padding_side": "right",20        "padding": "max_length",21        "truncation": true,22        "tokenizer_name": "facebook/bart-large"23      }24    },25    {26      "registry_name": "xvla_add_domain_id",27      "config": {28        "domain_id": 029      }30    },31    {32      "registry_name": "xvla_image_to_float",33      "config": {34        "image_keys": null,35        "validate_range": true36      }37    },38    {39      "registry_name": "xvla_imagenet_normalize",40      "config": {41        "image_keys": null42      }43    },44    {45      "registry_name": "device_processor",46      "config": {47        "device": "cuda",48        "float_dtype": null49      }50    },51    {52      "registry_name": "normalizer_processor",53      "config": {54        "eps": 1e-08,55        "features": {56          "observation.images.image": {57            "type": "VISUAL",58            "shape": [59              3,60              256,61              25662            ]63          },64          "observation.images.image2": {65            "type": "VISUAL",66            "shape": [67              3,68              256,69              25670            ]71          },72          "observation.state": {73            "type": "STATE",74            "shape": [75              876            ]77          },78          "observation.images.empty_camera_0": {79            "type": "VISUAL",80            "shape": [81              3,82              224,83              22484            ]85          },86          "action": {87            "type": "ACTION",88            "shape": [89              2090            ]91          }92        },93        "norm_map": {94          "STATE": "IDENTITY",95          "ACTION": "MEAN_STD",96          "VISUAL": "IDENTITY"97        }98      }99    }100  ]101}