From b939a1e5acd4e8b85c2549ca20c5131a0a220491 Mon Sep 17 00:00:00 2001
From: Brett Kuprel <brkuprel@gmail.com>
Date: Sun, 3 Jul 2022 12:20:03 -0400
Subject: [PATCH] update cog.yaml

---
 cog.yaml | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/cog.yaml b/cog.yaml
index 6b1401c..b69fc84 100644
--- 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"