s23deepak commited on
Commit
6aae577
·
verified ·
1 Parent(s): b7cd7a3

Add setup.py

Browse files
Files changed (1) hide show
  1. setup.py +30 -0
setup.py ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ """Setup for StreamingVLM Qwen3-VL port."""
2
+
3
+ from setuptools import setup, find_packages
4
+
5
+ setup(
6
+ name="streaming-vlm-qwen3",
7
+ version="0.1.0",
8
+ description="StreamingVLM: Real-time video understanding with Qwen3-VL on ROCm",
9
+ author="PitchAI",
10
+ packages=find_packages(),
11
+ python_requires=">=3.10",
12
+ install_requires=[
13
+ "torch>=2.4.0",
14
+ "torchvision>=0.19.0",
15
+ "accelerate>=1.2.0",
16
+ "qwen-vl-utils>=0.0.14",
17
+ "Pillow>=10.0.0",
18
+ "numpy>=1.24.0",
19
+ "tqdm",
20
+ ],
21
+ extras_require={
22
+ "video": ["decord>=0.6.0", "av>=12.0.0"],
23
+ "train": ["deepspeed>=0.14.0", "peft>=0.13.0", "liger-kernel>=0.5.0", "datasets>=2.0.0"],
24
+ },
25
+ entry_points={
26
+ "console_scripts": [
27
+ "streaming-vlm=streaming_vlm.inference.inference:main",
28
+ ],
29
+ },
30
+ )