updated replicate cog.yaml python packages

2 years ago · 31fd7c231d
parent 849e720313
commit 31fd7c231d
1 changed files with 1 additions and 2 deletions
--- 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"