config.json

{
  "architectures": [
    "TRENModel"
  ],
  "auto_map": {
    "AutoConfig": "configuration_tren.TRENConfig",
    "AutoModel": "modeling_tren.TRENModel"
  },
  "dtype": "float32",
  "hidden_dim": 1024,
  "image_resolution": 512,
  "merging_iou_threshold": 0.8,
  "merging_similarity_threshold": 0.975,
  "model_type": "tren",
  "num_attention_heads": 8,
  "num_decoder_layers": 2,
  "num_multiscale_regions": 3,
  "patch_size": 16,
  "text_embed_dim": 1024,
  "transformers_version": "5.2.0"
}