diff --git a/replicate_predictor.py b/replicate_predictor.py index 85b28a6..9b70e92 100644 --- a/replicate_predictor.py +++ b/replicate_predictor.py @@ -6,7 +6,7 @@ from cog import BasePredictor, Path, Input class ReplicatePredictor(BasePredictor): def setup(self): - self.model = MinDalle(is_mega=True) + self.model = MinDalle(is_mega=True, is_reusable=True) def predict( self,