tanner/min-dalle-test
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Update cog.yaml

Browse Source
This commit is contained in:
Brett Kuprel 2022-06-29 19:14:10 -04:00 committed by GitHub
parent 5f931f6899
commit 8b86152359
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
cog.yaml vendored
View File

@ -6,9 +6,7 @@ build:
- "libgl1-mesa-glx" - "libgl1-mesa-glx"
- "libglib2.0-0" - "libglib2.0-0"
python_packages: python_packages:
- "ipython==7.21.0"
- "torch==1.10.1" - "torch==1.10.1"
- "flax==0.4.2" - "flax==0.4.2"
- "wandb==0.12.16"
predict: "predict.py:Predictor" predict: "predict.py:Predictor"