| { | |
| "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" | |
| } | |