From acf3f8703b2afd2595bd352a9085adfa620e5147 Mon Sep 17 00:00:00 2001
From: zanussbaum <zanussbaum@gmail.com>
Date: Sat, 8 Apr 2023 17:03:30 -0400
Subject: [PATCH] fix: bs

---
 configs/train/finetune_gptj_lora.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/configs/train/finetune_gptj_lora.yaml b/configs/train/finetune_gptj_lora.yaml
index 1586b7e7..5d7e3fc0 100644
--- a/configs/train/finetune_gptj_lora.yaml
+++ b/configs/train/finetune_gptj_lora.yaml
@@ -9,7 +9,7 @@ streaming: false
 num_proc: 64
 dataset_path: "nomic-ai/turbo-500k-multi"
 max_length: 1024
-batch_size: 4
+batch_size: 2
 
 # train dynamics
 lr: 2.0e-5