update replicate files

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

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"

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)