Model save

README.md

@@ -16,7 +16,7 @@ model-index:
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 should probably proofread and complete it, then remove this comment. -->
 
-[<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="200" height="32"/>](https://wandb.ai/quirky_lats_at_mats/harmbench_adv_training/runs/3s3lr6jr)
+[<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="200" height="32"/>](https://wandb.ai/quirky_lats_at_mats/harmbench_adv_training/runs/jpr0o2in)
 # llama2-7b-sft-lora
 
 This model is a fine-tuned version of [meta-llama/Llama-2-7b-chat-hf](https://huggingface.co/meta-llama/Llama-2-7b-chat-hf) on the generator dataset.
@@ -43,10 +43,10 @@ The following hyperparameters were used during training:
 - eval_batch_size: 8
 - seed: 4
 - distributed_type: multi-GPU
-- num_devices: 4
+- num_devices: 8
 - gradient_accumulation_steps: 4
-- total_train_batch_size: 64
-- total_eval_batch_size: 32
+- total_train_batch_size: 128
+- total_eval_batch_size: 64
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: cosine
 - training_steps: 2000

adapter_config.json

@@ -16,12 +16,12 @@
   "rank_pattern": {},
   "revision": null,
   "target_modules": [
-    "q_proj",
+    "k_proj",
     "down_proj",
     "o_proj",
-    "k_proj",
     "up_proj",
-    "v_proj"
+    "v_proj",
+    "q_proj"
   ],
   "task_type": "CAUSAL_LM"
 }

runs/May21_08-25-05_d5a9515dbe64/events.out.tfevents.1716280080.d5a9515dbe64.9982.0

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

step_0/adapter_config.json

@@ -16,12 +16,12 @@
   "rank_pattern": {},
   "revision": null,
   "target_modules": [
-    "q_proj",
+    "k_proj",
     "down_proj",
     "o_proj",
-    "k_proj",
     "up_proj",
-    "v_proj"
+    "v_proj",
+    "q_proj"
   ],
   "task_type": "CAUSAL_LM"
 }

step_0/training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3ab92a3f3ddfae79ccc268f8c775b89b44bbf68264640ce3b02e1761ffa03dd0
+oid sha256:673be1a62de90ba9b79355d701aeedd980026a4e9778de4d5d7b1561b8c95e2b
 size 6200

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3ab92a3f3ddfae79ccc268f8c775b89b44bbf68264640ce3b02e1761ffa03dd0
+oid sha256:673be1a62de90ba9b79355d701aeedd980026a4e9778de4d5d7b1561b8c95e2b
 size 6200