pascal-maker commited on
Commit
1b26d88
·
verified ·
1 Parent(s): 9acc591

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +27 -23
requirements.txt CHANGED
@@ -1,30 +1,34 @@
1
- # Core ML/AI frameworks
2
- torch>=2.0.0
3
- numpy>=1.24.0
4
- Pillow>=9.0.0
 
 
5
 
6
- # Transformers and related
7
- transformers>=4.36.0
8
- accelerate>=0.20.0
9
- qwen-vl-utils>=0.0.8
10
 
11
- # Gradio for web interface
12
  gradio>=4.0.0
13
 
14
- # SAM-2 dependencies
15
- opencv-python>=4.8.0
16
- matplotlib>=3.5.0
17
- hydra-core>=1.3.0
18
- omegaconf>=2.3.0
 
19
 
20
- # Additional utilities
21
- safetensors>=0.3.0
22
- tokenizers>=0.13.0
23
- huggingface-hub>=0.19.0
24
- sentencepiece>=0.1.99
25
- protobuf>=3.20.0
26
 
27
- # For CheXagent streaming (threading is built into Python)
 
 
 
 
28
 
29
- # Optional: flash-attn for better performance (commented out due to installation complexity)
30
- # flash-attn>=2.0.0
 
1
+ ## Core ML and Image Processing
2
+ torch>=2.5.1
3
+ torchvision>=0.20.1
4
+ transformers>=4.30.0
5
+ numpy>=1.24.4
6
+ pillow>=9.4.0
7
 
8
+ # Computer Vision and Augmentation
9
+ opencv-python>=4.5.0
10
+ albumentations>=1.3.0
11
+ einops>=0.6.0
12
 
13
+ # Web Interface
14
  gradio>=4.0.0
15
 
16
+ # Utilities
17
+ tqdm>=4.66.1
18
+ hydra-core>=1.3.2
19
+ omegaconf>=2.2.0
20
+ pyyaml>=6.0
21
+ packaging>=21.0
22
 
23
+ # Additional dependencies that might be needed
24
+ iopath>=0.1.10
25
+ portalocker>=2.0.0
 
 
 
26
 
27
+ # Optional but recommended
28
+ matplotlib>=3.5.0
29
+ seaborn>=0.11.0
30
+ scikit-image>=0.19.0
31
+ scikit-learn>=1.0.0
32
 
33
+ # For better performance (optional)
34
+ accelerate>=0.20.0