diff --git a/replicate/model.py b/replicate/model.py index 4787e337..081386c4 100644 --- a/replicate/model.py +++ b/replicate/model.py @@ -23,11 +23,11 @@ class Model(BaseModel): def predict(self, *args, **kwargs) -> None: """ - DEPRECATED: Use `version.predict()` instead. + DEPRECATED: Use `replicate.run()` instead. """ raise ReplicateException( - "The `model.predict()` method has been removed, because it's unstable: if a new version of the model you're using is pushed and its API has changed, your code may break. Use `version.predict()` instead. See https://github.com/replicate/replicate-python#readme" + "The `model.predict()` method has been removed, because it's unstable: if a new version of the model you're using is pushed and its API has changed, your code may break. Use `replicate.run()` instead. See https://github.com/replicate/replicate-python#readme" ) @property diff --git a/replicate/version.py b/replicate/version.py index 71cee3ad..2f68f20f 100644 --- a/replicate/version.py +++ b/replicate/version.py @@ -32,6 +32,8 @@ class Version(BaseModel): def predict(self, **kwargs) -> Union[Any, Iterator[Any]]: """ + DEPRECATED: Use `replicate.run()` instead. + Create a prediction using this model version. Args: