| base_model: EleutherAI/pythia-1b | |
| batch_size: 32 | |
| bf16: true | |
| chat_template: tokenizer_default_fallback_alpaca | |
| datasets: | |
| - data_files: | |
| - b2a4966d9a5c880e_train_data.json | |
| ds_type: json | |
| format: custom | |
| path: /workspace/input_data/b2a4966d9a5c880e_train_data.json | |
| type: | |
| field_input: input | |
| field_instruction: instruction | |
| field_output: output | |
| 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/ee937811-31d0-4e11-944a-f4f8e06309d2 | |
| 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 | |
| special_tokens: | |
| pad_token: <|endoftext|> | |
| tokenizer_type: GPTNeoXTokenizerFast | |
| train_on_inputs: false | |
| trust_remote_code: true | |
| val_set_size: 0.1 | |
| wandb_entity: '' | |
| wandb_mode: online | |
| wandb_name: EleutherAI/pythia-1b-/workspace/input_data/b2a4966d9a5c880e_train_data.json | |
| wandb_project: Gradients-On-Demand | |
| wandb_run: your_name | |
| wandb_runid: default | |
| warmup_ratio: 0.05 | |
| xformers_attention: true | |