From 7b554e991f1499c3160d629444d4718d0334a9e1 Mon Sep 17 00:00:00 2001
From: Ivan Lee <84584280+ivanleekk@users.noreply.github.com>
Date: Sun, 17 Aug 2025 20:44:52 +0800
Subject: [PATCH] feat: main slurm

---
 main.slurm | 14 ++++++++++++++
 1 file changed, 14 insertions(+)
 create mode 100644 main.slurm

diff --git a/main.slurm b/main.slurm
new file mode 100644
index 00000000..150e604e
--- /dev/null
+++ b/main.slurm
@@ -0,0 +1,14 @@
+#!/bin/bash
+#SBATCH --job-name=vllm_test
+#SBATCH --output=vllm_test%j.out
+#SBATCH --error=vllm_test%j.err
+#SBATCH --time=01:00:00
+#SBATCH --partition=gpu
+#SBATCH --ntasks=1
+#SBATCH --cpus-per-task=4
+#SBATCH --gpus=h100:1
+#SBATCH --mem=80G
+
+# Run the script
+uv sync
+uv run main.py