update cog.yaml

This commit is contained in:
Brett Kuprel 2022-07-04 10:06:09 -04:00
parent deefd24919
commit 486c73c741

2
cog.yaml vendored
View File

@ -6,7 +6,7 @@ build:
- "libgl1-mesa-glx"
- "libglib2.0-0"
python_packages:
- "min-dalle==0.2.16"
- "min-dalle==0.2.17"
run:
- pip install torch==1.10.0+cu113 -f https://download.pytorch.org/whl/torch_stable.html