Update requirements.txt

requirements.txt

@@ -1,47 +1,17 @@
-# Common
-einops==0.7.0                   # Tensor operations
-
-# Training
-torch==2.4.0                    # Torch
-omegaconf==2.3.0                # Configuration system.
-lpips==0.1.4                    # 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
-
-# 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
-
-# Metrics
-torchmetrics==1.3.2             # Core module for metric
-pycocotools==2.0.7              # COCO-related
-torch-fidelity==0.3.0           # 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
-
-# Testing
-parameterized==0.9.0            # Define multiple tests through decorators.
-
-# Notebook
-mediapy==1.2.0                  # 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
+--extra-index-url https://download.pytorch.org/whl/cu128
+torch==2.8.0
+torchvision==0.23.0
+safetensors
+numpy
+tqdm
+psutil
+einops
+omegaconf>=2.3.0
+diffusers>=0.33.1
+peft>=0.17.0
+rotary_embedding_torch>=0.5.3
+opencv-python-headless
+gguf
+matplotlib
+mediapy
+huggingface_hub