config.json

{
  "model_type": "diffretriever",
  "architectures": [
    "DiffRetrieverModel"
  ],
  "auto_map": {
    "AutoConfig": "configuration_diffretriever.DiffRetrieverConfig",
    "AutoModel": "modeling_diffretriever.DiffRetrieverModel"
  },
  "base_model": "GSAI-ML/LLaDA-8B-Instruct",
  "backbone_type": "llada1",
  "mode": "single",
  "k_q": 1,
  "k_p": 1
}