shouryap commited on
Commit
759c277
·
1 Parent(s): 24da929
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
- torch==2.4.1 # 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
 
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