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

update replicate files

Browse Source
This commit is contained in:
Brett Kuprel 2022-07-05 05:47:35 -04:00
parent ccdcbc7d46
commit 45cf9cf5a7
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
cog.yaml vendored
View File

@ -10,4 +10,4 @@ build:
run: run:
- pip install torch==1.10.0+cu113 -f https://download.pytorch.org/whl/torch_stable.html - pip install torch==1.10.0+cu113 -f https://download.pytorch.org/whl/torch_stable.html
predict: "cogrun.py:Predictor" predict: "replicate_predictor.py:ReplicatePredictor"

2
cogrun.py → replicate_predictor.py
View File

@ -6,7 +6,7 @@ from math import log2
from cog import BasePredictor, Path, Input from cog import BasePredictor, Path, Input
class Predictor(BasePredictor): class ReplicatePredictor(BasePredictor):
def setup(self): def setup(self):
self.model = MinDalle(is_mega=True) self.model = MinDalle(is_mega=True)