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-07-03 12:20:03 -04:00
parent f0f9c5e83e
commit b939a1e5ac
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
cog.yaml vendored
View File

@ -7,5 +7,7 @@ build:
- "libglib2.0-0"
python_packages:
- "min-dalle==0.2.10"
run:
- pip install torch==1.10.0+cu113 -f https://download.pytorch.org/whl/torch_stable.html
predict: "replicate/predict.py:Predictor"