lstm_v1 / config.json
AlexHung29629's picture
Upload LstmForCausalLM
d7a28b9 verified
raw
history blame contribute delete
581 Bytes
{
"_name_or_path": "/home/alex/Workspace/LSTM_LM/lstm_v1/checkpoint-2000",
"architectures": [
"LstmForCausalLM"
],
"auto_map": {
"AutoConfig": "configuration_lstm.LstmConfig",
"AutoModelForCausalLM": "modeling_lstm.LstmForCausalLM"
},
"bos_token_id": 128000,
"eos_token_id": 128001,
"hidden_size": 2048,
"initializer_gain": 1,
"initializer_range": 0.02,
"intermediate_size": 8192,
"model_type": "lstm",
"num_hidden_layers": 2,
"pad_token_id": 128004,
"torch_dtype": "bfloat16",
"transformers_version": "4.47.1",
"vocab_size": 128256
}