Hugging Face's logo

ZhankuiHe
/
unicrs-gen

Transformers
Safetensors
Inference Endpoints
Model card Files Community
unicrs-gen
File size: 822 Bytes 
f4cad71
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
 
{
  "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"
}