Hello Google Cloud Community,
I’m using gemini-1.5-pro-001 for some classification pipelines using vertexai and I’m getting the following error in one of my prober tests –
ValueError: Unknown model publishers/google/models/gemini-1.5-pro-001; {'gs://google-cloud-aiplatform/schema/predict/instance/text_embedding_1.0.0.yaml': <class 'vertexai.preview.language_models._PreviewTextEmbeddingModel'>}.
What’s interesting is that this error is only occurring when I’m changing the deployment region to ‘euwest1’ using –
if constants.DEPLOYMENT_REGION == "EU":
deployment_region = constants.EU_DEPLOYMENT_REGION_CODE
base_logger.info("Deployment region is EU, using %s", deployment_region)
try:
vertexai.init(location=deployment_region)
base_logger.info("Vertex AI initialized in %s", deployment_region)
# if the initialization fails in europe-west1 for any reason, terminate the process
except Exception as e:
base_logger.error(
"Failed to initialize Vertex AI in %s, aborting process",
deployment_region,
extra={"error": str(e)},
)
raise
Could anyone provide insights into what might be causing this issue or how to resolve it? Happy to provide more information in case you need!
Thank you for your assistance!