From 0353072d15029f721d320daa01bc4b628131a77f Mon Sep 17 00:00:00 2001
From: Starbeamrainbowlabs <sbrl@starbeamrainbowlabs.com>
Date: Fri, 4 Nov 2022 17:02:07 +0000
Subject: [PATCH] allow pretrain to run on gpu we've slashed the size of the
 2nd encoder, so ti should fit naow?

---
 aimodel/slurm-pretrain.job | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/aimodel/slurm-pretrain.job b/aimodel/slurm-pretrain.job
index 817c205..7c92810 100755
--- a/aimodel/slurm-pretrain.job
+++ b/aimodel/slurm-pretrain.job
@@ -5,7 +5,7 @@
 #SBATCH --gres=gpu:1
 #SBATCH -o %j.%N.%a.out.log
 #SBATCH -e %j.%N.%a.err.log
-#SBATCH -p gpu05
+#SBATCH -p gpu05,gpu
 #SBATCH --time=5-00:00:00
 #SBATCH --mem=61440
 # 61440 = 60GiB memory required