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 2e4dc9109bcba2571ac3340eff78f2748bc01d3b..9faf5e56bcc82c18abcbefdca31248a8f7bd2531 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
@@ -28,6 +28,9 @@ git status
 
 cd $SCRATCH_DIR/$DOWNLOAD_NAME
 
+export OMP_NUM_THREADS=1
+export PYTHONUNBUFFERED=1
+
 cp $SCRIPT_DIR/generate_hyperparameters.py .
 python generate_hyperparameters.py > hyperparameters.yaml
 
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 34556ac1a4b994604f145230b4388334c2942f38..a5199e860049702dd73f9f5b78dcd396d0c93d81 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
@@ -1,5 +1,5 @@
 #!/bin/bash
-#BSUB -J mhcf-{work_item_num} # Job name
+#BSUB -J MHCf-{work_item_num} # Job name
 #BSUB -P acc_nkcancer # allocation account or Unix group
 #BSUB -q normal # queue
 #BSUB -R span[hosts=1] # one node
@@ -16,6 +16,12 @@ export TMPDIR=/local/JOBS/mhcflurry-{work_item_num}
 export PATH=$HOME/.conda/envs/tf-cpu/bin/:$PATH
 export PYTHONUNBUFFERED=1
 
+# See https://www.tensorflow.org/guide/performance/overview
+export KMP_AFFINITY=granularity=fine,verbose,compact,1,0
+export KMP_SETTINGS=1
+export KMP_BLOCKTIME=0
+export OMP_NUM_THREADS=16
+
 set -e
 free -m