File size: 607 Bytes
9794be0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
#!/bin/bash
#SBATCH --job-name=text6
#SBATCH --nodes=1 # Request 1 node
#SBATCH --ntasks-per-node=1
#SBATCH --gpus-per-task=a100-40g:1
#SBATCH --cpus-per-gpu=8 # Number of CPU cores (threads) per task
#SBATCH --mem-per-gpu=40G # Memory limit per CPU core (there is no --mem-per-task)
#SBATCH --time=24:00:00 # Job timeout
#SBATCH --output=badprompt2_20250510_new_v1.log # Redirect stdout to a log file
cd /home/yuqian_fu/Projects/LLaVA
srun --nodes "$SLURM_NNODES" --ntasks-per-node 1 -- \
mkenv -f llava.yml -- \
sh -c "
python run_llava_ours.py
"
|