ViSS-8B / code /docs /INSTALL.md
mis2's picture
Release ViSS-8B weights, processors and source code
78ac017 verified
|
Raw History Blame Contribute Delete
2.03 kB

Installation

Use Python 3.10 or later. Run commands from the source root; editable installation keeps the configuration and script paths available.

CPU source checks

python -m venv .venv
source .venv/bin/activate
python -m pip install --upgrade pip
python -m pip install -e ".[test]"
python -m pytest -q tests

These tests exercise code contracts with synthetic inputs. They do not validate a trained checkpoint or execute a full benchmark.

GPU inference

Install a CUDA-enabled PyTorch build appropriate for the machine first, then install the inference extras:

python -m pip install -e ".[inference]"

The inference extras pin PyTorch 2.7.0 and Transformers 4.57.3. The PyTorch version follows the repository runtime requirements; the Transformers version follows the retained ViSS model configuration. The extras declare Accelerate, qwen-vl-utils, and Decord as well. They specify an installation recipe, not a claim that every GPU/driver combination has been validated.

Install FFmpeg and ensure ffprobe is on PATH. Multi-GPU shell launchers also use Bash 4 or newer, jq, sha256sum, and nvidia-smi.

Training and attention

python -m pip install -e ".[inference,training]"
python -m pip install flash-attn==2.7.4.post1 --no-build-isolation

FlashAttention installation needs a compatible CUDA toolkit and compiler. The trainer defaults to flash_attention_2. Use --attn-implementation sdpa for a development run when FlashAttention is unavailable; this changes the runtime configuration.

The package includes no model weights. Before a GPU run, confirm that the selected checkpoint loads with AutoModelForImageTextToText and that the processor matches the intended evaluation contract.

Environment records

Save the installed package versions and launch command alongside your own runs. Generated outputs belong under outputs/ or checkpoints/ and are excluded from the source archive. Do not commit credentials, benchmark media, or local machine paths.