aaurelions commited on
Commit
31d9ed0
·
verified ·
1 Parent(s): a699336

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +6 -7
requirements.txt CHANGED
@@ -1,10 +1,9 @@
1
  torch>=2.1.0 --index-url https://download.pytorch.org/whl/cpu
2
- transformers>=4.40.0
3
  accelerate>=0.25.0
4
- # bitsandbytes # Not strictly needed for CPU
5
  peft>=0.10.0
6
- trl>=0.8.0
7
- gradio>=4.20.0 # Try a more recent version
8
- sentencepiece
9
- protobuf
10
- numpy==1.26.4
 
1
  torch>=2.1.0 --index-url https://download.pytorch.org/whl/cpu
2
+ transformers[sentencepiece]>=4.40.0 # Added sentencepiece here
3
  accelerate>=0.25.0
4
+ # bitsandbytes # Optional for CPU, can remove if causing issues
5
  peft>=0.10.0
6
+ # trl>=0.8.0 # Not strictly needed for inference app, only for training
7
+ gradio>=4.15.0 # Try a version known to be stable or newer if buttons are desired
8
+ numpy==1.26.4
9
+ protobuf # Often a dependency for tokenizers