From 31fd7c231dcb72a5379c8d107cafc9ff91867c89 Mon Sep 17 00:00:00 2001
From: Brett Kuprel <brkuprel@gmail.com>
Date: Sat, 2 Jul 2022 11:23:32 -0400
Subject: [PATCH] updated replicate cog.yaml python packages

---
 cog.yaml | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/cog.yaml b/cog.yaml
index d835b67..82d01c7 100644
--- a/cog.yaml
+++ b/cog.yaml
@@ -6,7 +6,6 @@ build:
     - "libgl1-mesa-glx"
     - "libglib2.0-0"
   python_packages:
-    - "torch==1.10.1"
-    - "flax==0.5.2"
+    - "min-dalle>=0.2.10"
 
 predict: "replicate/predict.py:Predictor"