Update requirements.txt

requirements.txt

@@ -1,7 +1,7 @@
-torch==2.5.1+cu118  # Upgrade to a version that supports `float8_e4m3fn`
-torchvision==0.19.1  # Upgrade to match the PyTorch version
+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==1.2.1
+accelerate
 transformers
 scipy
 opencv-python
@@ -9,7 +9,6 @@ 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
+numpy<2  # Downgrade NumPy to avoid compatibility issues
+fastapi==0.112.2
+gradio==4.44.1