diff --git a/python/kserve/kserve/api/kserve_client.py b/python/kserve/kserve/api/kserve_client.py index 82fb2863eeb..8fe11069fca 100644 --- a/python/kserve/kserve/api/kserve_client.py +++ b/python/kserve/kserve/api/kserve_client.py @@ -422,6 +422,8 @@ def wait_model_ready(self, service_name, model_name, isvc_namespace=None, # pyl time.sleep(polling_interval) # Check model health API url = f"http://{cluster_ip}/{protocol_version}/models/{model_name}" + if protocol_version.lower() == "v2": + url = f"http://{cluster_ip}/{protocol_version}/models/{model_name}/ready" response = requests.get(url, headers=headers).status_code if response == 200: return