File
Browse files- requirements.txt +1 -1
requirements.txt
CHANGED
@@ -1,4 +1,4 @@
|
|
1 |
-
torch==2.
|
2 |
torchvision==0.16.2 # Ensure compatibility with the torch version
|
3 |
transformers==4.33.0 # Stable version for Hugging Face Transformers
|
4 |
addict==2.4.0 # For PyTorch utilities
|
|
|
1 |
+
torch==2.1.2 # Adjust CUDA version as needed (e.g., cu117 or cpu)
|
2 |
torchvision==0.16.2 # Ensure compatibility with the torch version
|
3 |
transformers==4.33.0 # Stable version for Hugging Face Transformers
|
4 |
addict==2.4.0 # For PyTorch utilities
|