update replicate files
This commit is contained in:
parent
ccdcbc7d46
commit
45cf9cf5a7
2
cog.yaml
vendored
2
cog.yaml
vendored
|
@ -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"
|
||||
|
|
|
@ -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)
|
||||
|
Loading…
Reference in New Issue
Block a user