Update requirements.txt

requirements.txt

@@ -1,47 +1,47 @@
 # Common
-einops==0.7.0                   # Tensor operations
+einops                  # Tensor operations
 
 # Training
-torch==2.4.0                    # Torch
-omegaconf==2.3.0                # Configuration system.
-lpips==0.1.4                    # LPIPS model for VAE training
+torch                 # Torch
+omegaconf                # Configuration system.
+lpips                   # LPIPS model for VAE training
 
 # Dataloading
-bson==0.5.10                    # bson for Image dataloading
-tensorflow==2.16.1              # Video dataloading
-opencv-python==4.9.0.80         # OpenCV
-sentencepiece==0.2.0            # For Text Encoder
+bson                # bson for Image dataloading
+tensorflow           # Video dataloading
+opencv-python         # OpenCV
+sentencepiece          # For Text Encoder
 
 # Modeling
-diffusers==0.29.1               # Diffusers
-rotary-embedding-torch==0.5.3   # Rotary positional embedding
-transformers==4.38.2            # Transformers
-torchvision==0.19.0             # Torchvision
-tiktoken==0.7.0                 # Tiktoken for generation
-transformers_stream_generator==0.0.5 #LLM generation support
+diffusers              # Diffusers
+rotary-embedding-torch   # Rotary positional embedding
+transformers         # Transformers
+torchvision          # Torchvision
+tiktoken               # Tiktoken for generation
+transformers_stream_generator #LLM generation support
 
 # Metrics
-torchmetrics==1.3.2             # Core module for metric
-pycocotools==2.0.7              # COCO-related
-torch-fidelity==0.3.0           # FID-related
+torchmetrics           # Core module for metric
+pycocotools           # COCO-related
+torch-fidelity           # FID-related
 
 # Experiment Tracking
-moviepy==1.0.3                  # WandB Logging Image & Video
-imageio==2.34.0                 # WandB Logging Image & Video
-tabulate==0.9.0                 # Logging Table
-deepdiff==7.0.1                 # Find difference of config
+moviepy                # WandB Logging Image & Video
+imageio                # WandB Logging Image & Video
+tabulate               # Logging Table
+deepdiff               # Find difference of config
 
 # Testing
-parameterized==0.9.0            # Define multiple tests through decorators.
+parameterized          # Define multiple tests through decorators.
 
 # Notebook
-mediapy==1.2.0                  # Notebook Visualization
+mediapy               # Notebook Visualization
 
 # DevOPs
-black >= 24, < 25               # Code formatting
-flake8 >= 7, < 8                # Code style
-isort >= 5, < 6                 # Import sorting
-pre-commit==3.7.0               # Pre-commit hooks
-expecttest==0.2.1               # Pytorch dist tests
-hypothesis==6.100.1             # Fix randomness
-av==12.0.0                      # Process audio and video
+black           # Code formatting
+flake8                # Code style
+isort                 # Import sorting
+pre-commit              # Pre-commit hooks
+expecttest             # Pytorch dist tests
+hypothesis           # Fix randomness
+av                   # Process audio and video