From 45cf9cf5a7655bf19d0614f991c9c63591e26729 Mon Sep 17 00:00:00 2001
From: Brett Kuprel <brkuprel@gmail.com>
Date: Tue, 5 Jul 2022 05:47:35 -0400
Subject: [PATCH] update replicate files

---
 cog.yaml                            | 2 +-
 cogrun.py => replicate_predictor.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)
 rename cogrun.py => replicate_predictor.py (95%)

diff --git a/cog.yaml b/cog.yaml
index 63ad6a4..46558c1 100644
--- 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"
diff --git a/cogrun.py b/replicate_predictor.py
similarity index 95%
rename from cogrun.py
rename to replicate_predictor.py
index 0697bf2..eccea66 100644
--- a/cogrun.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)