kimhyunwoo commited on
Commit
0ef9d3d
·
verified ·
1 Parent(s): 2673193

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +7 -11
requirements.txt CHANGED
@@ -1,12 +1,8 @@
1
- # Base libraries for model handling and UI
2
- transformers>=4.38.0,<4.41.0 # Specify a recent stable range
3
- torch>=2.0.0 # PyTorch (CPU version will be installed by HF)
4
- gradio>=4.20.0 # Gradio for the web interface
5
 
6
- # Helper libraries
7
- accelerate>=0.25.0 # Assists with model loading, even on CPU
8
- sentencepiece # Often required for tokenizers like this one
9
- protobuf # Sometimes a dependency for tokenizers/models
10
-
11
- # Optional: Consider if needed based on specific warnings during build
12
- # bitsandbytes --only-binary :all: # Usually for GPU quantization, likely not needed for CPU float32
 
1
+ # requirements.txt
 
 
 
2
 
3
+ torch --index-url https://download.pytorch.org/whl/cpu
4
+ transformers
5
+ gradio
6
+ sentencepiece
7
+ accelerate
8
+ huggingface-hub