main.py

git clone https://github.com/DLCV-BUAA/TinyLLaVABench.git
cd TinyLLaVABench

conda create -n tinyllava python=3.10 -y
conda activate tinyllava
pip install --upgrade pip  # enable PEP 660 support
pip install -e .

pip install -e ".[train]"
pip install flash-attn --no-build-isolation

git pull
pip install -e .