Wasim0606 commited on
Commit
09e0b6d
·
verified ·
1 Parent(s): a22d2ee

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -2
requirements.txt CHANGED
@@ -2,7 +2,9 @@
2
  huggingface_hub==0.25.2
3
  hf-transfer>=0.1.4
4
  datasets
5
- unsloth # Added for optimized Hugging Face model training
 
 
6
 
7
  # Transformers & LangChain
8
  transformers
@@ -12,7 +14,7 @@ chromadb
12
  # Model acceleration
13
  xformers
14
  accelerate>=0.26.0
15
- torch --index-url https://download.pytorch.org/whl/cu118 # Ensures GPU support
16
 
17
  # Audio processing
18
  soundfile
 
2
  huggingface_hub==0.25.2
3
  hf-transfer>=0.1.4
4
  datasets
5
+
6
+ # Optimized Hugging Face model training
7
+ unsloth
8
 
9
  # Transformers & LangChain
10
  transformers
 
14
  # Model acceleration
15
  xformers
16
  accelerate>=0.26.0
17
+ torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121 # CUDA 12.1 for your GPU
18
 
19
  # Audio processing
20
  soundfile