From f082c436b85e227ae946921c834411010c43b336 Mon Sep 17 00:00:00 2001
From: Ivan lee <ivan.lee@u.nus.edu>
Date: Sun, 17 Aug 2025 20:47:38 +0800
Subject: [PATCH] fix: update gpu config

---
 main.slurm | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/main.slurm b/main.slurm
index 150e604e..b5bed699 100644
--- a/main.slurm
+++ b/main.slurm
@@ -6,8 +6,8 @@
 #SBATCH --partition=gpu
 #SBATCH --ntasks=1
 #SBATCH --cpus-per-task=4
-#SBATCH --gpus=h100:1
-#SBATCH --mem=80G
+#SBATCH --gpus=h100-96:1
+#SBATCH --mem=96G
 
 # Run the script
 uv sync