| { |
| "WEIGHT_DIMENSIONS": { |
| "edge_index.dtype": "torch.int64", |
| "edge_index.shape": [ |
| 2, |
| 186685 |
| ], |
| "edge_type.dtype": "torch.int64", |
| "edge_type.shape": [ |
| 186685 |
| ] |
| }, |
| "_name_or_path": "recwizard/unicrs-rec-redial", |
| "architectures": [ |
| "UnicrsRec" |
| ], |
| "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_rec": 10, |
| "num_bases": 8, |
| "num_tokens": 50266, |
| "token_hidden_size": 768 |
| }, |
| "num_tokens": 50259, |
| "pad_token_id": 50257, |
| "pretrained_model": "microsoft/DialoGPT-small", |
| "torch_dtype": "float32", |
| "transformers_version": "4.33.2" |
| } |
|
|
