torch==2.5.1
torchaudio==2.4.1
torchvision==0.19.1
diffusers==0.31.0
accelerate
transformers
scipy
opencv-python
omegaconf
einops
roma
jaxtyping
numpy<2  # Downgrade NumPy to avoid compatibility issues