v0.2.0, MinDalleTorch -> MinDalle, breaking change

This commit is contained in:
Brett Kuprel
2022-07-01 19:44:24 -04:00
parent 2080e596c3
commit 35e97768a5
10 changed files with 43 additions and 45 deletions

View File

@@ -1,11 +1,11 @@
import tempfile
from cog import BasePredictor, Path, Input
from min_dalle.min_dalle_torch import MinDalleTorch
from min_dalle import MinDalle
class Predictor(BasePredictor):
def setup(self):
self.model = MinDalleTorch(is_mega=True)
self.model = MinDalle(is_mega=True)
def predict(
self,