Update requirements.txt

requirements.txt

@@ -1,7 +1,7 @@
-torch>=2.1.0  # Upgrade to a version that supports `float8_e4m3fn`
-torchvision>=0.16.0  # Upgrade to match the PyTorch version
+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
+accelerate==1.2.1
 transformers
 scipy
 opencv-python
@@ -9,6 +9,6 @@ omegaconf
 einops
 roma
 jaxtyping
-numpy<2  # Downgrade NumPy to avoid compatibility issues
+numpy==1.26.4  # Downgrade NumPy to avoid compatibility issues
 gradio==4.44.1
 gradio-client==1.3.0