FWKV-ROSA / config.json
FlameF0X's picture
Update config.json
7ed6b41 verified
Raw
History Blame Contribute Delete
654 Bytes
{
"accum_steps": 4,
"auto_map": {
"AutoModelForCausalLM": "modeling_fwkv.FWKVLanguageModel"
},
"architectures": [
"FWKVLanguageModel"
],
"batch_size": 8,
"ce_chunk": 512,
"d_emb": 128,
"d_model": 512,
"dataset_name": "HuggingFaceH4/ultrachat_200k",
"dtype": "float32",
"epochs": 2,
"ffn_mult": 4,
"grad_clip": 1.0,
"lr": 0.0003,
"max_train_examples": 20000,
"max_val_examples": 1000,
"model_type": "fwkv",
"n_layers": 14,
"seq_len": 1024,
"tie_word_embeddings": true,
"train_split": "train_sft",
"transformers_version": "5.0.0",
"val_split": "test_sft",
"vocab_size": 50259,
"wkv_floor": 0.1
}