submit_train.sh

#!/bin/bash
#SBATCH --job-name=qlora_train
#SBATCH --time=08:00:00
#SBATCH --gres=gpu:1
#SBATCH --mem=24G
#SBATCH --cpus-per-task=4
#SBATCH --output=qlora_train_%j.log

source ~/project/.venv/.venv/bin/activate
cd ~/project/m4_final_assignment
python Train_qlora_mistral_v2.PY