mistral-7b-mj-finetuned-HorrorStory / training_params.json
SayedNabeel's picture
Upload folder using huggingface_hub
a531969
raw
history blame
1.09 kB
{"model": "bn22/Mistral-7B-Instruct-v0.1-sharded", "data_path": ".", "project_name": "mistral-7b-mj-finetuned-HorrorStory", "train_split": "train", "valid_split": null, "text_column": "text", "rejected_text_column": "rejected", "lr": 0.0002, "epochs": 8, "batch_size": 8, "warmup_ratio": 0.1, "gradient_accumulation": 1, "optimizer": "adamw_torch", "scheduler": "linear", "weight_decay": 0.0, "max_grad_norm": 1.0, "seed": 42, "add_eos_token": false, "block_size": -1, "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": false, "push_to_hub": true, "use_int8": false, "model_max_length": 1024, "repo_id": "SayedNabeel/mistral-7b-mj-finetuned-HorrorStory", "use_int4": true, "trainer": "sft", "target_modules": "q_proj,v_proj", "merge_adapter": false, "username": null, "use_flash_attention_2": false, "log": "none", "disable_gradient_checkpointing": false, "model_ref": null, "dpo_beta": 0.1, "prompt_text_column": "prompt"}