| torch==2.5.1+cu118 # Upgrade to a version that supports `float8_e4m3fn` | |
| torchvision==0.19.1 # Upgrade to match the PyTorch version | |
| diffusers==0.31.0 # Pin a compatible version of diffusers | |
| accelerate==1.2.1 | |
| transformers | |
| scipy | |
| opencv-python | |
| omegaconf | |
| einops | |
| roma | |
| jaxtyping | |
| numpy==1.26.4 # Downgrade NumPy to avoid compatibility issues | |
| gradio==4.44.1 | |
| gradio-client==1.3.0 | |
| fastapi==0.112.2 |