luanns commited on
Commit
8729d55
·
verified ·
1 Parent(s): 893da74

Upload requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +39 -0
requirements.txt ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # GUI-Shift Requirements
2
+ # Core dependencies for VLM training and evaluation
3
+
4
+ # Deep Learning & Transformers
5
+ torch>=2.1.0
6
+ transformers>=4.40.0
7
+ trl>=0.13.0
8
+ accelerate>=0.28.0
9
+ deepspeed>=0.14.0
10
+
11
+ # Vision & Multimodal
12
+ qwen-vl-utils
13
+ Pillow>=10.0.0
14
+
15
+ # Data Processing
16
+ datasets>=2.18.0
17
+ numpy>=1.24.0
18
+ jsonlines>=4.0.0
19
+
20
+ # Training Utilities
21
+ wandb>=0.16.0
22
+ trackio>=0.5.0
23
+
24
+ # VLM-R1 Framework (install from GitHub)
25
+ # git+https://github.com/om-ai-lab/VLM-R1.git
26
+
27
+ # Evaluation
28
+ scikit-learn>=1.4.0
29
+
30
+ # Utilities
31
+ tqdm>=4.66.0
32
+ python-dotenv>=1.0.0
33
+ babel>=2.15.0
34
+ levenshtein>=0.25.0
35
+ json-repair>=0.20.0
36
+ math-verify>=0.2.0
37
+
38
+ # Optional: Flash Attention (use hub kernels instead)
39
+ # flash-attn>=2.5.0