feat: main slurm
This commit is contained in:
parent
9e32a416ed
commit
7b554e991f
|
|
@ -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
|
||||
Loading…
Reference in New Issue