#!/bin/bash
#SBATCH --job-name=activation
#SBATCH --gres=gpu:1
#SBATCH --cpus-per-task=8
#SBATCH --partition=physical-gpu
#SBATCH --mem=32G
#SBATCH --output=logs/%j-activation.out

ENV_FILE=$1

if [ -z "$ENV_FILE" ]; then
    echo "Error: No environment file specified"
    exit 1
fi

if [ ! -f "$ENV_FILE" ]; then
    echo "Error: Environment file $ENV_FILE not found"
    exit 1
fi

# Load the specified environment file
source "$ENV_FILE"

# Create the artifact folder
mkdir -p "$OUTPUT_DIR"

export PYTHONUNBUFFERED=1
export CUDA_VISIBLE_DEVICES=0
export VLLM_USE_V1=0
# Run the Python script
python ../activation_v2.py \
    --model $MODEL \
    --lang $LANG

echo "Activation single job done"