diff --git a/submit_experiment_batch.py b/submit_experiment_batch.py
index 3a1c92c..64bf3a5 100755
--- a/submit_experiment_batch.py
+++ b/submit_experiment_batch.py
@@ -69,7 +69,7 @@ def create_job_script(task, config, seed, phase):
     script_content = f'''#!/bin/bash
 #SBATCH --job-name=fasttd3_{phase}_{task.replace("-", "_")}_s{seed}
 #SBATCH --account=hk-project-p0022232
-#SBATCH --partition=accelerated
+#SBATCH --partition=accelerated-h100
 #SBATCH --time={config["time_limit"]}
 #SBATCH --gres=gpu:1
 #SBATCH --ntasks=1