base_model: /workspace/pythia-6.9b-deduped-8k/ base_model_config: EleutherAI/pythia-6.9b-deduped model_type: GPTNeoXForCausalLM tokenizer_type: AutoTokenizer load_in_8bit: false load_in_4bit: false strict: False gptq: false device_map: auto push_dataset_to_hub: winglian hf_use_auth_token: true # dataset_shard_num: 3 # dataset_shard_idx: 0 datasets: - path: winglian/pygmalion-cleaned data_files: - v13_no_ai.cleaned.jsonl type: pygmalion - path: winglian/evals data_files: - hf/ARC-Challenge.jsonl - hf/ARC-Easy.jsonl - hf/riddle_sense.jsonl type: explainchoice:chat - path: winglian/evals data_files: - openai/tldr.jsonl type: summarizetldr:chat - path: winglian/evals data_files: - hf/gsm8k.jsonl type: alpaca_chat.load_qa - path: winglian/evals data_files: - hellaswag/hellaswag.jsonl type: explainchoice:chat - path: metaeval/ScienceQA_text_only type: concisechoice:chat - path: ehartford/WizardLM_alpaca_evol_instruct_70k_unfiltered type: alpaca_chat - path: ehartford/wizard_vicuna_70k_unfiltered type: sharegpt:chat - path: winglian/chatlogs-en-cleaned data_files: - sharegpt_cleaned.jsonl type: sharegpt:chat - path: teknium/GPT4-LLM-Cleaned type: alpaca_chat - path: teknium/GPTeacher-General-Instruct data_files: gpt4-instruct-similarity-0.6-dataset.json type: gpteacher:chat - path: ewof/code-alpaca-instruct-unfiltered type: alpaca_chat - path: QingyiSi/Alpaca-CoT data_files: - Chain-of-Thought/formatted_cot_data/aqua_train.json - Chain-of-Thought/formatted_cot_data/creak_train.json - Chain-of-Thought/formatted_cot_data/ecqa_train.json - Chain-of-Thought/formatted_cot_data/esnli_train.json - Chain-of-Thought/formatted_cot_data/gsm8k_train.json - Chain-of-Thought/formatted_cot_data/qasc_train.json - Chain-of-Thought/formatted_cot_data/qed_train.json - Chain-of-Thought/formatted_cot_data/sensemaking_train.json - Chain-of-Thought/formatted_cot_data/strategyqa_train.json - GPTeacher/Roleplay/formatted_roleplay-similarity_0.6-instruct-dataset.json type: alpaca_chat dataset_prepared_path: last_run_prepared val_set_size: 0.01 adapter: lora_model_dir: sequence_len: 8192 max_packed_sequence_len: 8192 lora_r: 64 lora_alpha: 32 lora_dropout: 0.0 lora_target_modules: lora_target_linear: true lora_fan_in_fan_out: true # pythia/GPTNeoX lora specific wandb_project: minotaur-7b wandb_watch: wandb_run_id: wandb_log_model: output_dir: ./minotaur-7b gradient_accumulation_steps: 1 micro_batch_size: 4 num_epochs: 1 learning_rate: 0.00003 optimizer: adamw_bnb_8bit lr_scheduler: cosine warmup_steps: 100 save_steps: 84 eval_steps: 10 train_on_inputs: false group_by_length: true bf16: false fp16: false bfloat16: true tf32: true flash_optimum: true early_stopping_patience: resume_from_checkpoint: local_rank: gradient_checkpointing: true fsdp: fsdp_transformer_layer_cls_to_wrap: collator_pad_to_longest: true save_steps: 35