Skip to content

Fix vision engine build #60

Fix vision engine build

Fix vision engine build #60

Workflow file for this run

name: Unit Tests
on: [push, pull_request]
concurrency:
# run only one unit test for a branch / tag.
group: ci-${{ github.ref }}
cancel-in-progress: true
jobs:
lint:
runs-on: ubuntu-latest
steps:
- name: Check out Git repository
uses: actions/checkout@v4
- name: Set up Python
uses: actions/setup-python@v5
with:
python-version: '3.10'
- name: Install Python dependencies
run: |
# change to a single installation bash.
pip install black
pip install --upgrade pip # enable PEP 660 support
wget https://github.com/Dao-AILab/flash-attention/releases/download/v2.4.2/flash_attn-2.4.2+cu118torch2.0cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
pip install flash_attn-2.4.2+cu118torch2.0cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
pip install -e .
pip install -e ".[train]"
pip install pytorchvideo
pip install decord
pip install datasets
pip install scikit-learn
pip install openai
pip install git+https://github.com/huggingface/[email protected]
export PPATH=$(python -c 'import site; print(site.getsitepackages()[0])')
cp -rv ./llava/train/transformers_replace/* $PPATH/transformers/models/
- name: Run tests
run: |
#TODO(ligeng): switch to pytest
for f in tests/*.py; do
python $f
done