compliantLLM / config.json
Martin Navrátil
Upload 11 files
4689b4d verified
Raw
History Blame Contribute Delete
590 Bytes
{
"architectures": [
"CompliantLLMModel"
],
"auto_map": {
"AutoConfig": "configuration_compliant_llm.CompliantLLMConfig",
"AutoModel": "modeling_compliant_llm.CompliantLLMModel"
},
"d_model": 64,
"dropout": 0.0,
"ffn_dim": 128,
"input_vocab_size": 256,
"max_context": 1024,
"model_type": "compliant_llm",
"n_heads": 4,
"n_layers": 2,
"output_length": 3,
"output_tokens": [
"Sorry, but that question violates GDPR.",
"<|end_turn|>",
"<|eos|>"
],
"output_vocab_size": 3,
"torch_dtype": "float32",
"transformers_version": "5.14.1"
}