初始化项目,由ModelHub XC社区提供模型
Model: eekay/gemma-2b-it-steer-dog-numbers-ft-single-l13 Source: Original Platform
This commit is contained in:
46
config.json
Normal file
46
config.json
Normal file
@@ -0,0 +1,46 @@
|
||||
{
|
||||
"architectures": [
|
||||
"GemmaForCausalLM"
|
||||
],
|
||||
"attention_bias": false,
|
||||
"attention_dropout": 0.0,
|
||||
"bos_token_id": 2,
|
||||
"dtype": "bfloat16",
|
||||
"eos_token_id": 1,
|
||||
"ft_cfg": {
|
||||
"bf16": true,
|
||||
"continue_final_message": true,
|
||||
"dataset_name": "eekay/gemma-2b-it-steer-dog-numbers",
|
||||
"gradient_accumulation_steps": 1,
|
||||
"learning_rate": 0.001,
|
||||
"logging_steps": 100,
|
||||
"lora_layers": 13,
|
||||
"lora_rank": 8,
|
||||
"lr_scheduler_type": "constant",
|
||||
"max_grad_norm": 1.0,
|
||||
"model_id": "google/gemma-2b-it",
|
||||
"model_save_name": "gemma-2b-it-steer-dog-numbers-ft-single-l13",
|
||||
"n_examples": 30000,
|
||||
"num_train_epochs": 1,
|
||||
"per_device_train_batch_size": 24,
|
||||
"train_attn": false
|
||||
},
|
||||
"head_dim": 256,
|
||||
"hidden_act": "gelu",
|
||||
"hidden_activation": null,
|
||||
"hidden_size": 2048,
|
||||
"initializer_range": 0.02,
|
||||
"intermediate_size": 16384,
|
||||
"max_position_embeddings": 8192,
|
||||
"model_type": "gemma",
|
||||
"num_attention_heads": 8,
|
||||
"num_hidden_layers": 18,
|
||||
"num_key_value_heads": 1,
|
||||
"pad_token_id": 0,
|
||||
"rms_norm_eps": 1e-06,
|
||||
"rope_scaling": null,
|
||||
"rope_theta": 10000.0,
|
||||
"transformers_version": "4.57.3",
|
||||
"use_cache": true,
|
||||
"vocab_size": 256000
|
||||
}
|
||||
Reference in New Issue
Block a user