Omarito101 commited on
Commit
4593b74
·
verified ·
1 Parent(s): e869b70

Update requirements.txt with version compatibility fixes

Browse files
Files changed (1) hide show
  1. requirements.txt +7 -2
requirements.txt CHANGED
@@ -1,5 +1,10 @@
 
 
1
  torch>=2.0.0
2
- transformers>=4.40.0
3
- peft>=0.10.0
4
  accelerate>=0.28.0
5
  safetensors>=0.4.0
 
 
 
 
1
+ # Required versions for Energy Document Classifier
2
+ # Tested and verified to work together
3
  torch>=2.0.0
4
+ transformers>=4.44.0,<4.58.0
5
+ peft>=0.12.0,<0.19.0
6
  accelerate>=0.28.0
7
  safetensors>=0.4.0
8
+
9
+ # Note: The tokenizer is loaded from the base model (meta-llama/Llama-3.1-8B)
10
+ # You need access to the Llama-3.1 model on Hugging Face