|
base_model: peft-internal-testing/tiny-dummy-qwen2 |
|
batch_size: 8 |
|
bf16: true |
|
chat_template: tokenizer_default_fallback_alpaca |
|
datasets: |
|
- format: custom |
|
path: argilla/databricks-dolly-15k-curated-en |
|
type: |
|
field_input: original-instruction |
|
field_instruction: original-instruction |
|
field_output: original-response |
|
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/test-repo |
|
hub_strategy: checkpoint |
|
learning_rate: 0.002 |
|
load_best_model_at_end: true |
|
logging_steps: 10 |
|
lr_scheduler: cosine |
|
max_steps: 1 |
|
micro_batch_size: 1 |
|
model_type: AutoModelForCausalLM |
|
num_epochs: 100 |
|
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: 8 |
|
tokenizer_type: Qwen2TokenizerFast |
|
train_on_inputs: false |
|
trust_remote_code: true |
|
val_set_size: 0.001 |
|
wandb_entity: '' |
|
wandb_mode: online |
|
wandb_name: peft-internal-testing/tiny-dummy-qwen2-argilla/databricks-dolly-15k-curated-en |
|
wandb_project: Gradients-On-Demand |
|
wandb_run: your_name |
|
wandb_runid: default |
|
warmup_ratio: 0.05 |
|
xformers_attention: true |
|
|