training_params.json

{"model": "meta-llama/Llama-2-7b-hf", "data_path": "data/", "project_name": "llama-2-7b-hf-text2image-prompts-Liege", "train_split": "train_updated", "valid_split": "valid_updated", "text_column": "Texte", "lr": 0.0002, "epochs": 15, "batch_size": 1, "warmup_ratio": 0.1, "gradient_accumulation": 4, "optimizer": "adamw_torch", "scheduler": "linear", "weight_decay": 0.01, "max_grad_norm": 1.0, "seed": 42, "add_eos_token": false, "block_size": 1024, "use_peft": true, "lora_r": 16, "lora_alpha": 32, "lora_dropout": 0.05, "logging_steps": -1, "evaluation_strategy": "epoch", "save_total_limit": 1, "save_strategy": "epoch", "auto_find_batch_size": false, "fp16": true, "push_to_hub": true, "use_int8": false, "model_max_length": 1024, "repo_id": "pierreguillou/llama-2-7b-hf-text2image-prompts-Liege", "use_int4": true, "trainer": "default", "target_modules": null, "merge_adapter": true, "username": null}