base_model: unsloth/mistral-7b-v0.3 | |
batch_size: 32 | |
bf16: true | |
chat_template: tokenizer_default_fallback_alpaca | |
datasets: | |
- data_files: | |
- ca0152973425c947_train_data.json | |
ds_type: json | |
format: custom | |
path: /workspace/input_data/ca0152973425c947_train_data.json | |
type: | |
field_input: code | |
field_instruction: func_name | |
field_output: docstring | |
format: '{instruction} {input}' | |
no_input_format: '{instruction}' | |
system_format: '{system}' | |
system_prompt: '' | |
eval_steps: 20 | |
flash_attention: true | |
gpu_memory_limit: 80GiB | |
gradient_checkpointing: true | |
group_by_length: true | |
hub_model_id: willtensora/5a2f5ce6-446b-4282-bb4d-9ee4e970231f | |
hub_strategy: checkpoint | |
learning_rate: 0.0002 | |
logging_steps: 10 | |
lr_scheduler: cosine | |
max_steps: 2500 | |
micro_batch_size: 4 | |
model_type: AutoModelForCausalLM | |
optimizer: adamw_bnb_8bit | |
output_dir: /workspace/axolotl/configs | |
pad_to_sequence_len: true | |
resize_token_embeddings_to_32x: false | |
sample_packing: false | |
save_steps: 40 | |
save_total_limit: 1 | |
sequence_len: 2048 | |
tokenizer_type: LlamaTokenizerFast | |
train_on_inputs: false | |
trust_remote_code: true | |
val_set_size: 0.1 | |
wandb_entity: '' | |
wandb_mode: online | |
wandb_name: unsloth/mistral-7b-v0.3-/tmp/ca0152973425c947_train_data.json | |
wandb_project: Gradients-On-Demand | |
wandb_run: your_name | |
wandb_runid: default | |
warmup_ratio: 0.05 | |
xformers_attention: true | |