config.json

{
  "attn_depth": 12,
  "attn_dim_head": 64,
  "attn_heads": 12,
  "attn_kwargs": {
    "attn_num_mem_kv": 4,
    "ff_glu": true
  },
  "coarse_pre_gateloop_depth": 2,
  "condition_on_text": true,
  "cross_attn_num_mem_kv": 4,
  "dim": 768,
  "dropout": 0.0,
  "fine_attn_depth": 2,
  "fine_attn_dim_head": 32,
  "fine_attn_heads": 8,
  "fine_cross_attend_text": true,
  "fine_pre_gateloop_depth": 2,
  "flash_attn": true,
  "gateloop_use_heinsen": false,
  "max_seq_len": 1500,
  "num_sos_tokens": 1,
  "pad_id": -1,
  "quads": false,
  "text_cond_with_film": false,
  "text_condition_cond_drop_prob": 0.0,
  "text_condition_model_types": "bge"
}