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

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +5 -4
requirements.txt CHANGED
@@ -2,9 +2,7 @@
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,7 +12,9 @@ chromadb
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
@@ -23,3 +23,4 @@ librosa
23
  # Web UI
24
  gradio==3.50
25
  groq
 
 
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
  # Model acceleration
13
  xformers
14
  accelerate>=0.26.0
15
+
16
+ # PyTorch with CUDA (Make sure to install it manually)
17
+ # torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 # REMOVE this line from requirements.txt
18
 
19
  # Audio processing
20
  soundfile
 
23
  # Web UI
24
  gradio==3.50
25
  groq
26
+