kayfahaarukku commited on
Commit
f174f96
·
verified ·
1 Parent(s): b7558ff

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +5 -19
requirements.txt CHANGED
@@ -1,19 +1,5 @@
1
- # Install PyTorch with CUDA support
2
- --index-url https://download.pytorch.org/whl/cu121
3
- torch>=2.0.0
4
- torchvision>=0.15.0
5
- torchaudio>=2.0.0
6
-
7
- # Transformers and related packages
8
- transformers>=4.36.0
9
- peft>=0.7.0
10
- accelerate>=0.25.0
11
- huggingface_hub>=0.19.0
12
-
13
- # BitsAndBytes with CUDA support
14
- bitsandbytes>=0.41.0
15
-
16
- # Additional dependencies for GPU support
17
- ninja
18
- flash-attn>=2.0.0 --no-build-isolation
19
- gradio
 
1
+ torch>=2.0.0
2
+ transformers>=4.36.0
3
+ peft>=0.7.0
4
+ accelerate>=0.25.0
5
+ bitsandbytes>=0.41.0