torch>=2.1.0  # Upgrade to a version that supports `float8_e4m3fn`
torchvision>=0.16.0  # Upgrade to match the PyTorch version
diffusers==0.31.0  # Pin a compatible version of diffusers
accelerate
transformers
scipy
opencv-python
omegaconf
einops
roma
jaxtyping
numpy<2  # Downgrade NumPy to avoid compatibility issues