Update requirements.txt
Browse files- requirements.txt +1 -0
requirements.txt
CHANGED
|
@@ -1,6 +1,7 @@
|
|
| 1 |
# Base libraries for Hugging Face models
|
| 2 |
transformers==4.30.0 # Hugging Face Transformers library for model management
|
| 3 |
torch==2.0.0 # PyTorch library for model training and inference
|
|
|
|
| 4 |
# or for TensorFlow
|
| 5 |
# tensorflow==2.12.0 # Uncomment if using TensorFlow
|
| 6 |
|
|
|
|
| 1 |
# Base libraries for Hugging Face models
|
| 2 |
transformers==4.30.0 # Hugging Face Transformers library for model management
|
| 3 |
torch==2.0.0 # PyTorch library for model training and inference
|
| 4 |
+
einops==0.6.0 # Einops for tensor operations and manipulation
|
| 5 |
# or for TensorFlow
|
| 6 |
# tensorflow==2.12.0 # Uncomment if using TensorFlow
|
| 7 |
|