update replicate files

2022-07-05 05:47:35 -04:00
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"