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