goodgoals commited on
Commit
11f6d37
·
verified ·
1 Parent(s): 3cf5419

Create requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +18 -0
requirements.txt ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Core ML Stack
2
+ torch
3
+ torchvision
4
+ torchaudio
5
+ --extra-index-url https://download.pytorch.org
6
+
7
+ # Unsloth & Performance
8
+ unsloth[colab-new] @ git+https://github.com
9
+ xformers
10
+ bitsandbytes
11
+ triton
12
+
13
+ # Training & Datasets
14
+ transformers
15
+ trl
16
+ accelerate
17
+ datasets
18
+ sentencepiece