update replicate files

2 years ago · 45cf9cf5a7
parent ccdcbc7d46
commit 45cf9cf5a7
2 changed files with 2 additions and 2 deletions
--- a/cog.yaml
+++ b/cog.yaml
@ -10,4 +10,4 @@ build:
  run:
    - 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"
--- a/replicate_predictor.py
+++ b/replicate_predictor.py
@ -6,7 +6,7 @@ from math import log2
 from cog import BasePredictor, Path, Input


-class Predictor(BasePredictor):
+class ReplicatePredictor(BasePredictor):
    def setup(self):
        self.model = MinDalle(is_mega=True)