text_encoder/config.json

{
  "_class_name": "AuroraTextEncoder",
  "_diffusers_version": "0.42.1",
  "bos_id": 2,
  "depth": 6,
  "dim": 384,
  "eos_id": 3,
  "head_dim": 64,
  "heads": 6,
  "kv_heads": 2,
  "mask_id": 5,
  "max_tokens": 64,
  "mlp_hidden": 1024,
  "null_id": 4,
  "pad_id": 0,
  "vocab_size": 8192
}