update replicate_predictor.py

2022-07-05 07:07:29 -04:00
parent 9a4d9dc118
commit c96ba0805e
1 changed files with 1 additions and 1 deletions
--- 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,