diff --git a/downloads-generation/models_class1_pan_unselected/GENERATE.WITH_HPC_CLUSTER.sh b/downloads-generation/models_class1_pan_unselected/GENERATE.WITH_HPC_CLUSTER.sh
index 0a064dfebf82bc26fe70ba161554af6a7e515e40..848010267a95868d94b188a31fd0748be38d180b 100755
--- a/downloads-generation/models_class1_pan_unselected/GENERATE.WITH_HPC_CLUSTER.sh
+++ b/downloads-generation/models_class1_pan_unselected/GENERATE.WITH_HPC_CLUSTER.sh
@@ -45,8 +45,8 @@ do
         --verbosity 1 \
         --cluster-parallelism \
         --cluster-submit-command bsub \
-        --cluster-results-workdir ~/mhcflurry-model-training \
-        --cluster-script-prefix-path ./cluster_submit_script_header.mssm_hpc.lsf
+        --cluster-results-workdir ~/mhcflurry-scratch \
+        --cluster-script-prefix-path $SCRIPT_DIR/cluster_submit_script_header.mssm_hpc.lsf
 done
 
 cp $SCRIPT_ABSOLUTE_PATH .
diff --git a/downloads-generation/models_class1_pan_unselected/cluster_submit_script_header.mssm_hpc.lsf b/downloads-generation/models_class1_pan_unselected/cluster_submit_script_header.mssm_hpc.lsf
index d8d9342eaec044935ba7708a0b8c778e6241028c..2844b3135603c7019335dc3244f19e205e5bb4e6 100644
--- a/downloads-generation/models_class1_pan_unselected/cluster_submit_script_header.mssm_hpc.lsf
+++ b/downloads-generation/models_class1_pan_unselected/cluster_submit_script_header.mssm_hpc.lsf
@@ -10,10 +10,14 @@
 #BSUB -eo {work_dir}/%J.stderr # error log
 #BSUB -L /bin/bash # Initialize the execution environment
 
-export TMPDIR=/local/JOBS/haplocnv-{case}
-export PATH=$HOME/.conda/envs/py37/bin/:$PATH
+export TMPDIR=/local/JOBS/mhcflurry-{work_item_num}
+export PATH=$HOME/.conda/envs/py36b/bin/:$PATH
 
 set -e
 free -m
+
+module load cuda/10.1.105
+module load cudnn/7.1.1
+
 cd {work_dir}