use torch 2.4.1

cog.yaml

@@ -7,8 +7,8 @@ build:
   python_version: "3.10.14"
   python_packages:
     # flux
-    - "torch==2.4.0"
-    - "torchvision==0.19.0"
+    - "torch==2.4.1"
+    - "torchvision==0.19.1"
     - "safetensors==0.4.4"
     - "diffusers==0.30.0"
     - "transformers==4.44.0"
@@ -56,8 +56,8 @@ build:
   run:
     - curl -o /usr/local/bin/pget -L "https://github.com/replicate/pget/releases/download/v0.8.2/pget_linux_x86_64" && chmod +x /usr/local/bin/pget
     - pip install "git+https://github.com/Gothos/diffusers.git@flux-inpaint"
-    - "pip install --force-reinstall --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu124"
-    #- "pip install --force-reinstall --pre torch==2.5.0.dev20240905+cu124 torchvision==0.20.0.dev20240905+cu124 --index-url https://download.pytorch.org/whl/nightly/cu124"
+    # - "pip install --force-reinstall --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu124"
+    # - "pip install --force-reinstall --pre torch==2.5.0.dev20240905+cu124 torchvision==0.20.0.dev20240905+cu124 --index-url https://download.pytorch.org/whl/nightly/cu124"
 
 # predict.py defines how predictions are run on your model
 predict: "predict.py:Predictor"