config.json

{
  "activation": "gelu",
  "arch": "trans_enc",
  "condition": "text",
  "dropout": 0.1,
  "ff_size": 1024,
  "flip_sin_to_cos": true,
  "freq_shift": 0,
  "guidance_scale": 7.5,
  "guidance_uncondp": 0.1,
  "latent_dim": [
    1,
    256
  ],
  "nclasses": 10,
  "nfeats": 263,
  "normalize_before": false,
  "num_heads": 4,
  "num_layers": 9,
  "pe_type": "mld",
  "position_embedding": "learned",
  "return_intermediate_dec": false,
  "skip_connect": true,
  "text_encoded_dim": 768
}