update cog.yaml

2 years ago · b939a1e5ac
parent f0f9c5e83e
commit b939a1e5ac
1 changed files with 2 additions and 0 deletions
--- a/cog.yaml
+++ b/cog.yaml
@ -7,5 +7,7 @@ build:
    - "libglib2.0-0"
  python_packages:
    - "min-dalle==0.2.10"
+  run:
+    - pip install torch==1.10.0+cu113 -f https://download.pytorch.org/whl/torch_stable.html

 predict: "replicate/predict.py:Predictor"