Update requirements.txt with version compatibility fixes
Browse files- requirements.txt +7 -2
requirements.txt
CHANGED
|
@@ -1,5 +1,10 @@
|
|
|
|
|
|
|
|
| 1 |
torch>=2.0.0
|
| 2 |
-
transformers>=4.
|
| 3 |
-
peft>=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
|