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

update replicate_predictor.py

Browse Source
This commit is contained in:
Brett Kuprel
2022-07-05 07:07:29 -04:00
parent 9a4d9dc118
commit c96ba0805e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
replicate_predictor.py
View File

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