Faaz commited on
Commit
24b5fb1
·
1 Parent(s): 4a33f96

Add requirements-training.txt for MI300X Docker

Browse files
Files changed (1) hide show
  1. requirements-training.txt +26 -0
requirements-training.txt ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # MINDI 1.5 — Training requirements (MI300X / Python 3.10+)
2
+ # torch is pre-installed in the Docker image — do NOT include here
3
+
4
+ # Core ML
5
+ transformers>=4.40.0
6
+ peft>=0.10.0
7
+ accelerate>=0.29.0
8
+ datasets>=2.19.0
9
+ safetensors>=0.4.0
10
+ tokenizers>=0.19.0
11
+
12
+ # Vision
13
+ Pillow>=10.0.0
14
+ open-clip-torch>=2.24.0
15
+
16
+ # HuggingFace + Logging
17
+ huggingface_hub>=0.23.0
18
+ wandb>=0.17.0
19
+
20
+ # Config + Utils
21
+ PyYAML>=6.0
22
+ python-dotenv>=1.0.0
23
+ tqdm>=4.66.0
24
+ numpy>=1.24.0
25
+ regex>=2024.0.0
26
+ psutil>=5.9.0