diff --git a/aimodel/slurm-pretrain-predict.job b/aimodel/slurm-pretrain-predict.job
index 1ee7d23..4fe26b1 100755
--- a/aimodel/slurm-pretrain-predict.job
+++ b/aimodel/slurm-pretrain-predict.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
 
 module load utilities/multi