config.json

{
  "dim": 1280,
  "n_layer": 30,
  "n_head": 20,
  "n_kv_head": null,
  "multiple_of": 256,
  "ffn_dim_multiplier": null,
  "rope_base": 10000,
  "norm_eps": 1e-05,
  "initializer_range": 0.02,
  "token_dropout_p": 0.1,
  "attn_dropout_p": 0.0,
  "resid_dropout_p": 0.1,
  "ffn_dropout_p": 0.1,
  "drop_path_rate": 0.0,
  "num_classes": 101,
  "class_dropout_prob": 0.1,
  "model_type": "class_cond",
  "vocab_size": 8192,
  "cls_token_num": 1025,
  "max_batch_size": 32,
  "max_seq_len": 1536,
  "use_fixed_pe": false,
  "frame_prediction": true,
  "mode": "train_adap_ar_fp",
  "with_block_sep": true,
  "tot_blocks": 3,
  "custom_cls_token_num": true,
  "start_ar_block_idx": 1
}