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/4u15rhza)
+[<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/adv-robustness/harmbench_adv_training/runs/gh4yjslx)
 # 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

@@ -17,11 +17,11 @@
   "revision": null,
   "target_modules": [
     "v_proj",
+    "o_proj",
     "k_proj",
-    "q_proj",
+    "down_proj",
     "up_proj",
-    "o_proj",
-    "down_proj"
+    "q_proj"
   ],
   "task_type": "CAUSAL_LM"
 }

runs/May20_22-42-48_9962c5574dd5/events.out.tfevents.1716245043.9962c5574dd5.30217.0

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

step_0/adapter_config.json

@@ -17,11 +17,11 @@
   "revision": null,
   "target_modules": [
     "v_proj",
+    "o_proj",
     "k_proj",
-    "q_proj",
+    "down_proj",
     "up_proj",
-    "o_proj",
-    "down_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:3d5a2bc4755dd43d7d40ba9e853eb1528b6f82c1de5eaf623707ae7753c17001
+oid sha256:4f2f622bba93f7fa7370ea5bb89ea9e60affba5b860979b74173ce836a22fde8
 size 6200

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3d5a2bc4755dd43d7d40ba9e853eb1528b6f82c1de5eaf623707ae7753c17001
+oid sha256:4f2f622bba93f7fa7370ea5bb89ea9e60affba5b860979b74173ce836a22fde8
 size 6200