feat: main slurm

This commit is contained in:
Ivan Lee 2025-08-17 20:44:52 +08:00
parent 9e32a416ed
commit 7b554e991f
1 changed files with 14 additions and 0 deletions

14
main.slurm Normal file
View File

@ -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