citywalker / config.json
Gaaaavin's picture
v2: full encoder weights + auto_map for trust_remote_code=True
cec7d2d verified
{
"architectures": [
"CityWalkerModel"
],
"auto_map": {
"AutoConfig": "configuration_citywalker.CityWalkerConfig",
"AutoModel": "modeling_citywalker.CityWalkerModel"
},
"context_size": 5,
"cord_include_input": true,
"cord_num_freqs": 6,
"crop": [
400,
400
],
"decoder_ff_dim_factor": 4,
"decoder_num_heads": 8,
"decoder_num_layers": 16,
"do_resize": true,
"do_rgb_normalize": true,
"dtype": "float32",
"freeze_obs_encoder": true,
"len_traj_pred": 5,
"model_type": "citywalker",
"obs_encoder_type": "dinov2_vitb14",
"resize": [
392,
392
],
"transformers_version": "5.8.0"
}