{ "architectures": [ "RetroGPTForCausalLM" ], "model_type": "retrogpt", "vocab_size": 439, "d_model": 512, "n_heads": 8, "n_layers": 6, "num_hidden_layers": 6, "hidden_size": 512, "num_attention_heads": 8, "max_seq_len": 256, "dropout": 0.1, "sep_id": 3, "pad_token_id": 0, "bos_token_id": 1, "eos_token_id": 2, "auto_map": { "AutoConfig": "modeling_retrogpt.RetroGPTConfig", "AutoModelForCausalLM": "modeling_retrogpt.RetroGPTForCausalLM", "AutoTokenizer": [ "modeling_retrogpt.RetroGPTTokenizer", null ] } }