kimhyunwoo commited on
Commit
6d73a79
·
verified ·
1 Parent(s): ca39c1f

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +12 -6
requirements.txt CHANGED
@@ -1,6 +1,12 @@
1
- transformers>=4.38.0 # Use a recent version
2
- torch>=2.0.0 # PyTorch CPU version
3
- gradio>=4.0.0 # Gradio for the UI
4
- accelerate>=0.25.0 # Helps with model loading, even on CPU
5
- sentencepiece # Often needed for tokenizers
6
- protobuf # Sometimes needed as a dependency by tokenizers/models
 
 
 
 
 
 
 
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