updated replicate cog.yaml python packages

This commit is contained in:
Brett Kuprel 2022-07-02 11:23:32 -04:00
parent 849e720313
commit 31fd7c231d

3
cog.yaml vendored
View File

@ -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"