andreamena commited on
Commit
36731e3
·
verified ·
1 Parent(s): b050433

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +6 -34
requirements.txt CHANGED
@@ -1,35 +1,7 @@
1
- # PyTorch CUDA 12.6 stack
2
- --extra-index-url https://download.pytorch.org/whl/cu126
3
- torch==2.7.0
4
- torchvision==0.22.0
5
- torchaudio==2.7.0
6
-
7
- # Core ML/Training libraries
8
- transformers==4.53.2
9
- accelerate==1.9.0
10
- datasets==3.6.0
11
- trl
12
- peft==0.16.0
13
  smact
14
-
15
- # Quantization & optimization
16
- bitsandbytes==0.46.1
17
- xformers==0.0.30
18
-
19
- # Unsloth for fast fine-tuning
20
- unsloth==2025.6.12
21
- unsloth_zoo==2025.7.6
22
-
23
- # Materials science (CrysText specific)
24
- pymatgen==2025.6.14
25
-
26
- # Experiment tracking
27
- wandb==0.21.0
28
-
29
- # Essential utilities (auto-installed as dependencies, but pinned for stability)
30
- numpy==2.2.6
31
- pandas==2.3.1
32
- scipy==1.16.0
33
- tokenizers==0.21.2
34
- safetensors==0.5.3
35
- huggingface-hub==0.33.4
 
1
+ unsloth
2
+ torch
3
+ transformers
4
+ pymatgen
 
 
 
 
 
 
 
 
5
  smact
6
+ gradio
7
+ numpy