File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -429,7 +429,7 @@ def optimize(
429429
430430 _OPTIMIZER_METHOD_TO_CONTAINER_URI = {
431431 types .PromptOptimizerMethod .VAPO : "us-docker.pkg.dev/vertex-ai/cair/vaipo:preview_v1_0" ,
432- types .PromptOptimizerMethod .OPTIMIZATION_TARGET_GEMINI_NANO : "us-docker.pkg.dev/vertex-ai/cair/android-apo:preview_v1_0 " ,
432+ types .PromptOptimizerMethod .OPTIMIZATION_TARGET_GEMINI_NANO : "us-docker.pkg.dev/vertex-ai/cair/vaipo:preview_android_v1_0 " ,
433433 }
434434 container_uri = _OPTIMIZER_METHOD_TO_CONTAINER_URI .get (method )
435435 if not container_uri :
@@ -809,7 +809,7 @@ async def optimize(
809809
810810 _OPTIMIZER_METHOD_TO_CONTAINER_URI = {
811811 types .PromptOptimizerMethod .VAPO : "us-docker.pkg.dev/vertex-ai/cair/vaipo:preview_v1_0" ,
812- types .PromptOptimizerMethod .OPTIMIZATION_TARGET_GEMINI_NANO : "us-docker.pkg.dev/vertex-ai/cair/android-apo:preview_v1_0 " ,
812+ types .PromptOptimizerMethod .OPTIMIZATION_TARGET_GEMINI_NANO : "us-docker.pkg.dev/vertex-ai/cair/vaipo:preview_android_v1_0 " ,
813813 }
814814 container_uri = _OPTIMIZER_METHOD_TO_CONTAINER_URI .get (method )
815815 if not container_uri :
You can’t perform that action at this time.
0 commit comments