AxionLab-official commited on
Commit
9000f03
·
verified ·
1 Parent(s): fd17ee1

Create requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +28 -0
requirements.txt ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # ──────────────────────────────────────────────────
2
+ # Fine-Tuning Studio — requirements.txt
3
+ # Compatible: CPU / CPU Upgrade / T4 / A10G / A100
4
+ # ──────────────────────────────────────────────────
5
+
6
+ # Core ML
7
+ torch>=2.2.0
8
+ transformers>=4.40.0
9
+ datasets>=2.19.0
10
+ accelerate>=0.29.0
11
+ tokenizers>=0.19.0
12
+
13
+ # Fine-tuning eficiente (LoRA / QLoRA)
14
+ peft>=0.10.0
15
+ bitsandbytes>=0.43.0 # GPU only — não falha em CPU (graceful import)
16
+
17
+ # UI
18
+ gradio>=4.31.0
19
+
20
+ # Data
21
+ pandas>=2.0.0
22
+ pyarrow>=14.0.0
23
+
24
+ # Utils
25
+ safetensors>=0.4.3
26
+ huggingface_hub>=0.23.0
27
+ sentencepiece>=0.2.0 # alguns tokenizers precisam
28
+ protobuf>=3.20.0