Update requirements.txt

requirements.txt

@@ -2,7 +2,7 @@
 einops                  # Tensor operations
 
 # Training
-torch                 # Torch
+torch==2.4.0                 # Torch
 omegaconf                # Configuration system.
 lpips                   # LPIPS model for VAE training
 
@@ -16,7 +16,7 @@ sentencepiece          # For Text Encoder
 diffusers              # Diffusers
 rotary-embedding-torch   # Rotary positional embedding
 transformers         # Transformers
-torchvision          # Torchvision
+torchvision==0.19.0          # Torchvision
 tiktoken               # Tiktoken for generation
 transformers_stream_generator #LLM generation support