#!/usr/bin/env bash

WORKDIR="/var/task"
XDG_CACHE_HOME="/data"
MPLCONFIGDIR=${XDG_CACHE_HOME}/.cache/matplotlib
TRANSFORMERS_CACHE=${XDG_CACHE_HOME}/.cache/transformers

ls -ld ${XDG_CACHE_HOME}/
ls -l ${XDG_CACHE_HOME}/

mkdir -p ${XDG_CACHE_HOME}/.cache
chmod 770 -R ${XDG_CACHE_HOME}/.cache
mkdir -p ${MPLCONFIGDIR}
mkdir -p ${TRANSFORMERS_CACHE}

ls -ld ${XDG_CACHE_HOME}/
ls -l ${XDG_CACHE_HOME}/
export WORKDIR
export XDG_CACHE_HOME
export MPLCONFIGDIR
export TRANSFORMERS_CACHE

source ${WORKDIR}/venv/bin/activate

which python
python --version

free -m
which nvcc
nvcc -V
which nvidia-smi
nvidia-smi
pip list

df -h / /data /home /var/task

echo "WORKDIR - /var/task::"
ls -l ${WORKDIR}

echo "XDG_CACHE_HOME - /data"
find ${XDG_CACHE_HOME}

CUDA_VISIBLE_DEVICES=$(nvidia-smi --query-gpu=memory.free,index --format=csv,nounits,noheader | sort -nr | head -1 | awk '{ print $NF }')
echo "calculated CUDA_VISIBLE_DEVICES env variable: ${CUDA_VISIBLE_DEVICES}."
export CUDA_VISIBLE_DEVICES

python ${WORKDIR}/app.py --version='xinlai/LISA-13B-llama2-v1-explanatory' --precision='fp16' --load_in_4bit

exit 0