{ "architectures": [ "CustomModelForCausalLM" ], "auto_map": { "AutoConfig": "configuration_custom_mbz_test.CustomConfig", "AutoModelForCausalLM": "modeling_custom_mbz_test.CustomModelForCausalLM", "AutoModel": "modeling_custom_mbz_test.CustomModelForCausalLM" }, "d_head": 128, "d_model": 8192, "dtype": "float32", "model_type": "custom-mbz-test", "n_heads": 64, "n_layers": 3, "n_vocab": 50257, "transformers_version": "4.56.0" }