From d0f7e3903006b6b0b5f9e8e7bc09b55904cd5ce5 Mon Sep 17 00:00:00 2001 From: andreasjansson Date: Wed, 11 Sep 2024 15:38:25 +0200 Subject: [PATCH] use torch 2.4.1 --- cog.yaml | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/cog.yaml b/cog.yaml index eb3c084..d614d0b 100644 --- a/cog.yaml +++ b/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"