{
  "WEIGHT_DIMENSIONS": {
    "edge_index.dtype": "torch.int64",
    "edge_index.shape": [
      2,
      186685
    ],
    "edge_type.dtype": "torch.int64",
    "edge_type.shape": [
      186685
    ]
  },
  "_name_or_path": "../local_repo/unicrs-gen",
  "architectures": [
    "UnicrsGen"
  ],
  "kgprompt_config": {
    "hidden_size": 768,
    "kg_info": {
      "num_entities": 31162,
      "num_relations": 25,
      "pad_entity_id": 31161
    },
    "n_block": 2,
    "n_head": 12,
    "n_layer": 12,
    "n_prefix_conv": 20,
    "n_prefix_rec": null,
    "num_bases": 8,
    "num_tokens": 50266,
    "token_hidden_size": 768
  },
  "max_gen_len": 50,
  "num_tokens": 50259,
  "pad_token_id": 50257,
  "pretrained_model": "microsoft/DialoGPT-small",
  "torch_dtype": "float32",
  "transformers_version": "4.38.1"
}