allow setup(self, weights: str)

Signed-off-by: technillogue <technillogue@gmail.com>
replicate · Mar 5, 2024 · 81f9659 · 81f9659
1 parent 46d0eb3
commit 81f9659
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/python/cog/predictor.py b/python/cog/predictor.py
@@ -85,6 +85,9 @@ def run_setup(predictor: BasePredictor) -> None:
         elif weights_type == CogPath:
             # TODO: So this can be a url. evil!
             weights = cast(CogPath, CogPath.validate(weights_url))
+        # allow people to download weights themselves
+        elif weights_type == str:
+            return weights_url
         else:
             raise ValueError(
                 f"Predictor.setup() has an argument 'weights' of type {weights_type}, but only File and Path are supported"