jake2004 commited on
Commit
79f9aec
·
verified ·
1 Parent(s): 2a24c78

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +5 -8
requirements.txt CHANGED
@@ -1,17 +1,14 @@
1
  # Streamlit for Web UI
2
  streamlit
3
 
4
- # Hugging Face Hub & Inference Client
5
  huggingface_hub
6
-
7
- # Transformers for AI Model Loading
8
  transformers
9
 
10
- # Torch (GPU version - CUDA 11.8)
11
- torch==2.2.0+cu118 torchvision==0.17.0+cu118 torchaudio==2.2.0+cu118 --index-url https://download.pytorch.org/whl/cu118
12
-
13
- # Uncomment the below line if running on CPU-only
14
- # torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
15
 
16
  # OpenAI's accelerate for faster model inference
17
  accelerate
 
1
  # Streamlit for Web UI
2
  streamlit
3
 
4
+ # Hugging Face Hub & Transformers
5
  huggingface_hub
 
 
6
  transformers
7
 
8
+ # PyTorch (CPU-only version)
9
+ torch --index-url https://download.pytorch.org/whl/cpu
10
+ torchvision --index-url https://download.pytorch.org/whl/cpu
11
+ torchaudio --index-url https://download.pytorch.org/whl/cpu
 
12
 
13
  # OpenAI's accelerate for faster model inference
14
  accelerate