nqlm / config.json
tommytracx's picture
Add config.json
7b7d538 verified
{
"architectures": [
"NeuralQuantumNQLM"
],
"auto_map": {
"AutoModel": "modeling_nqlm.NeuralQuantumNQLMForCausalLM",
"AutoConfig": "configuration_nqlm.NeuralQuantumNQLMConfig"
},
"model_type": "neuralquantum_nqlm",
"vocab_size": 50257,
"hidden_size": 768,
"num_attention_heads": 12,
"num_hidden_layers": 12,
"intermediate_size": 3072,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.1,
"attention_probs_dropout_prob": 0.1,
"max_position_embeddings": 512,
"type_vocab_size": 2,
"initializer_range": 0.02,
"layer_norm_eps": 1e-12,
"use_cache": true,
"quantum_enhancement": true,
"quantum_layers": 4,
"quantum_circuit_depth": 8,
"quantum_optimization": "vqe",
"hybrid_mode": true,
"torch_dtype": "float16"
}