update cog.yaml

This commit is contained in:
Brett Kuprel
2022-07-03 16:16:23 -04:00
parent 948e05c0dd
commit 6ce60179ff
2 changed files with 4 additions and 3 deletions
Vendored
+1 -1
View File
@@ -6,7 +6,7 @@ build:
- "libgl1-mesa-glx"
- "libglib2.0-0"
python_packages:
- "min-dalle==0.2.10"
- "min-dalle==0.2.12"
run:
- pip install torch==1.10.0+cu113 -f https://download.pytorch.org/whl/torch_stable.html