Update cog.yaml

This commit is contained in:
Brett Kuprel 2022-06-29 19:14:25 -04:00 committed by GitHub
parent 8b86152359
commit c76226571e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

2
cog.yaml vendored
View File

@ -7,6 +7,6 @@ build:
- "libglib2.0-0" - "libglib2.0-0"
python_packages: python_packages:
- "torch==1.10.1" - "torch==1.10.1"
- "flax==0.4.2" - "flax==0.5.2"
predict: "predict.py:Predictor" predict: "predict.py:Predictor"