Update requirements.txt

requirements.txt

@@ -26,6 +26,11 @@ pydantic
 soundfile
 SeedVR2Pipeline
 
+tiktoken
+transformers_stream_generator
+
+rotary-embedding-torch
+
 # --- Data Handling & I/O ---
 Pillow
 numpy
@@ -40,4 +45,31 @@ av
 # --- Hugging Face Hub ---
 huggingface_hub
 
-google-generativeai
+google-generativeai
+
+
+# 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