| { | |
| "_name_or_path": "TForge_model/v2", | |
| "architectures": [ | |
| "TForgeModel" | |
| ], | |
| "decoder_start_token_id": 2, | |
| "dropout_prob": 0.1, | |
| "embed_dim": 512, | |
| "eos_token_id": 3, | |
| "forced_eos_token_id": 3, | |
| "hidden_dim": 1024, | |
| "input_vocab_size": 10000, | |
| "is_encoder_decoder": true, | |
| "load_pretrained": true, | |
| "max_positions": 256, | |
| "model_type": "Transformer_Forge", | |
| "num_heads": 8, | |
| "num_layers": 6, | |
| "output_vocab_size": 10000, | |
| "pad_token_id": 1, | |
| "soft_align_dim": 512, | |
| "torch_dtype": "float32", | |
| "transformers_version": "4.23.1", | |
| "translate_vocab_size": 10000 | |
| } | |