{ "model": "openai-community/gpt2", "project_name": "att-1d-test", "data_path": "att-1d-test/autotrain-data", "train_split": "train", "valid_split": null, "add_eos_token": true, "block_size": 128, "model_max_length": 512, "padding": "right", "trainer": "sft", "use_flash_attention_2": false, "log": "tensorboard", "disable_gradient_checkpointing": false, "logging_steps": -1, "evaluation_strategy": "epoch", "save_total_limit": 1, "auto_find_batch_size": false, "mixed_precision": "fp16", "lr": 2e-05, "epochs": 10, "batch_size": 32, "warmup_ratio": 0.1, "gradient_accumulation": 2, "optimizer": "adamw_torch", "scheduler": "linear", "weight_decay": 0.0, "max_grad_norm": 1.0, "seed": 42, "chat_template": null, "quantization": "int4", "target_modules": "all-linear", "merge_adapter": false, "peft": false, "lora_r": 16, "lora_alpha": 32, "lora_dropout": 0.05, "model_ref": null, "dpo_beta": 0.1, "max_prompt_length": 128, "max_completion_length": null, "prompt_text_column": "autotrain_prompt", "text_column": "autotrain_text", "rejected_text_column": "autotrain_rejected_text", "push_to_hub": true, "username": "BananaSauce" }