Training in progress, step 100

adapter_config.json

@@ -0,0 +1,26 @@
+{
+  "auto_mapping": null,
+  "base_model_name_or_path": "mistralai/Mistral-7B-v0.1",
+  "bias": "none",
+  "fan_in_fan_out": false,
+  "inference_mode": true,
+  "init_lora_weights": true,
+  "layers_pattern": null,
+  "layers_to_transform": null,
+  "lora_alpha": 32,
+  "lora_dropout": 0.1,
+  "modules_to_save": null,
+  "peft_type": "LORA",
+  "r": 64,
+  "revision": null,
+  "target_modules": [
+    "v_proj",
+    "o_proj",
+    "up_proj",
+    "gate_proj",
+    "down_proj",
+    "k_proj",
+    "q_proj"
+  ],
+  "task_type": "CAUSAL_LM"
+}

adapter_model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dda825c13ac7b2f848b970b524f518b6456014cb3e473ece417bae2eb6739174
+size 335605144

training_args.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1a00d251b9e8778867a0be2ae762151214e219b14ed22e794563e9fffee1931b
+size 6264

training_config.json

@@ -0,0 +1,86 @@
+{
+  "experiment_key": "base",
+  "save_safetensors": true,
+  "max_shard_size": "10GB",
+  "local_rank": 1,
+  "use_gradient_checkpointing": true,
+  "trainer_key": "lm",
+  "force_fp32": false,
+  "force_fp16": false,
+  "from_gptq": false,
+  "huggingface_hub_token": null,
+  "deepspeed_stage": 2,
+  "deepspeed_config_path": null,
+  "fsdp_strategy": "",
+  "fsdp_offload": true,
+  "seed": 42,
+  "stabilize": true,
+  "path_to_env_file": "./.env",
+  "prepare_dataset": true,
+  "lora_hub_model_id": null,
+  "lora_model_local_path": null,
+  "fused_model_local_path": null,
+  "quantization_dataset_id": null,
+  "quantization_max_samples": 1024,
+  "quantized_model_path": "./quantized_model/",
+  "quantized_hub_model_id": null,
+  "quantized_hub_private_repo": null,
+  "dataset_key": "desc2json",
+  "train_local_path_to_data": "./train.jsonl",
+  "eval_local_path_to_data": null,
+  "shuffle": true,
+  "max_eval_samples": 1000,
+  "add_eval_to_train_if_no_path": false,
+  "tokenizer_name_or_path": null,
+  "tokenizer_use_fast": null,
+  "tokenizer_padding_side": null,
+  "collator_key": "completion",
+  "max_length": 2048,
+  "model_name_or_path": "mistralai/Mistral-7B-v0.1",
+  "push_to_hub_bos_add_bos_token": false,
+  "use_flash_attention_2": false,
+  "trust_remote_code": true,
+  "device_map": null,
+  "prepare_model_for_kbit_training": true,
+  "load_in_8bit": false,
+  "load_in_4bit": true,
+  "llm_int8_threshold": 6.0,
+  "llm_int8_has_fp16_weight": true,
+  "bnb_4bit_use_double_quant": true,
+  "bnb_4bit_quant_type": "nf4",
+  "bnb_quantize_after_model_init": false,
+  "gptq_bits": 4,
+  "gptq_group_size": 128,
+  "gptq_disable_exllama": true,
+  "apply_lora": true,
+  "lora_rank": 64,
+  "lora_alpha": 32,
+  "lora_dropout": 0.1,
+  "raw_lora_target_modules": "all",
+  "output_dir": "./outputs/",
+  "per_device_train_batch_size": 2,
+  "do_eval": false,
+  "per_device_eval_batch_size": null,
+  "gradient_accumulation_steps": 4,
+  "eval_accumulation_steps": null,
+  "eval_delay": 0,
+  "eval_steps": 1000,
+  "warmup_steps": 100,
+  "max_steps": 5000,
+  "num_train_epochs": 3,
+  "learning_rate": 0.0002,
+  "max_grad_norm": 1.0,
+  "weight_decay": 0.001,
+  "label_smoothing_factor": 0.1,
+  "logging_steps": 1,
+  "save_steps": 100,
+  "save_total_limit": 0,
+  "optim": "paged_adamw_8bit",
+  "push_to_hub": true,
+  "hub_model_id": "BobaZooba/WGPT-LoRA",
+  "hub_private_repo": false,
+  "report_to_wandb": true,
+  "wandb_api_key": null,
+  "wandb_project": null,
+  "wandb_entity": null
+}