初始化项目,由ModelHub XC社区提供模型
Model: Pomilon/LEMA-llama-2-7b Source: Original Platform
This commit is contained in:
30
final_checkpoint/lema_config.json
Normal file
30
final_checkpoint/lema_config.json
Normal file
@@ -0,0 +1,30 @@
|
||||
{
|
||||
"model_name_or_path": "NousResearch/Llama-2-7b-hf",
|
||||
"model_type": null,
|
||||
"gbi_path": "llama2_7b.safetensors",
|
||||
"device": "cuda",
|
||||
"strategy": "streaming",
|
||||
"ram_buffer_size": 2,
|
||||
"vram_buffer_size": 1,
|
||||
"use_lora": true,
|
||||
"lora_rank": 16,
|
||||
"lora_alpha": 32,
|
||||
"lora_target_modules": [
|
||||
"q_proj",
|
||||
"k_proj",
|
||||
"v_proj",
|
||||
"o_proj",
|
||||
"gate_proj",
|
||||
"up_proj",
|
||||
"down_proj"
|
||||
],
|
||||
"learning_rate": 0.0001,
|
||||
"batch_size": 8,
|
||||
"gradient_accumulation_steps": 1,
|
||||
"max_seq_length": 512,
|
||||
"gradient_checkpointing": true,
|
||||
"save_steps": 500,
|
||||
"output_dir": "checkpoints",
|
||||
"dtype": "float16",
|
||||
"attn_implementation": "eager"
|
||||
}
|
||||
Reference in New Issue
Block a user