setup.py

"""Setup for StreamingVLM Qwen3-VL port."""

from setuptools import setup, find_packages

setup(
    name="streaming-vlm-qwen3",
    version="0.1.0",
    description="StreamingVLM: Real-time video understanding with Qwen3-VL on ROCm",
    author="PitchAI",
    packages=find_packages(),
    python_requires=">=3.10",
    install_requires=[
        "torch>=2.4.0",
        "torchvision>=0.19.0",
        "accelerate>=1.2.0",
        "qwen-vl-utils>=0.0.14",
        "Pillow>=10.0.0",
        "numpy>=1.24.0",
        "tqdm",
    ],
    extras_require={
        "video": ["decord>=0.6.0", "av>=12.0.0"],
        "train": ["deepspeed>=0.14.0", "peft>=0.13.0", "liger-kernel>=0.5.0", "datasets>=2.0.0"],
    },
    entry_points={
        "console_scripts": [
            "streaming-vlm=streaming_vlm.inference.inference:main",
        ],
    },
)