TengfeiLiuCoder commited on
Commit
403ebeb
·
verified ·
1 Parent(s): 39b517d

Update inference environment documentation

Browse files
Files changed (3) hide show
  1. README.md +4 -2
  2. requirement.txt +12 -0
  3. requirements.txt +0 -5
README.md CHANGED
@@ -33,8 +33,10 @@ The released weights are a merged Hugging Face checkpoint based on [Qwen3-VL-8B-
33
 
34
  ## Installation
35
 
 
 
36
  ```bash
37
- pip install -r requirements.txt
38
  ```
39
 
40
  RefCaptioner uses the standard Qwen3-VL inference stack and does not require a task-specific runtime. Any environment that can run Qwen3-VL with a compatible Transformers release can be used directly.
@@ -45,7 +47,7 @@ The benchmark environment used `torch==2.9.0`, `transformers==4.57.6`, `accelera
45
 
46
  ```bash
47
  python inference.py \
48
- --model NOVAglow646/RefCaptioner \
49
  --video /path/to/video.mp4 \
50
  --images /path/to/ref_1.jpg /path/to/ref_2.jpg /path/to/ref_3.png
51
  ```
 
33
 
34
  ## Installation
35
 
36
+ The tested environment for loading RefCaptioner and running inference is provided in `requirement.txt`. Use Python 3.10 and install it with:
37
+
38
  ```bash
39
+ pip install -r requirement.txt
40
  ```
41
 
42
  RefCaptioner uses the standard Qwen3-VL inference stack and does not require a task-specific runtime. Any environment that can run Qwen3-VL with a compatible Transformers release can be used directly.
 
47
 
48
  ```bash
49
  python inference.py \
50
+ --model TengfeiLiuCoder/RefCaptioner \
51
  --video /path/to/video.mp4 \
52
  --images /path/to/ref_1.jpg /path/to/ref_2.jpg /path/to/ref_3.png
53
  ```
requirement.txt ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Tested model inference environment for RefCaptioner (Python 3.10).
2
+ torch==2.9.0
3
+ torchvision==0.24.0
4
+ transformers==4.57.6
5
+ accelerate==1.14.0
6
+ qwen-vl-utils==0.0.14
7
+ av==17.1.0
8
+ pillow==12.2.0
9
+ numpy==2.2.6
10
+ peft==0.19.1
11
+ safetensors==0.8.0
12
+ einops==0.8.2
requirements.txt DELETED
@@ -1,5 +0,0 @@
1
- torch>=2.4
2
- transformers>=4.57.0,<5
3
- accelerate>=1.0
4
- qwen-vl-utils>=0.0.14
5
- av>=14