Upload source/requirements.txt with huggingface_hub

#32
Files changed (1) hide show
  1. source/requirements.txt +19 -0
source/requirements.txt ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Install with: pip install -r requirements.txt
2
+ # WARNING: Do NOT run pip install torch - NV custom build will be overwritten
3
+
4
+ # Core - DO NOT reinstall torch (NV custom build already installed)
5
+ # torch is pre-installed as NV build 2.10.0a0+nv25.12
6
+ # flash_attn is pre-installed as 2.7.4
7
+
8
+ transformers>=4.40.0
9
+ tokenizers>=0.19.0
10
+ datasets>=2.18.0
11
+ accelerate>=0.30.0
12
+ peft>=0.10.0
13
+ trl>=0.8.6
14
+ sentencepiece>=0.1.99
15
+ wandb>=0.16.0
16
+ tensorboard>=2.16.0
17
+ PyYAML>=6.0
18
+ numpy>=1.26.0
19
+ tqdm>=4.66.0