diff --git a/serverless/cloud-run-gpu/ollama-demo/__main__.py b/serverless/cloud-run-gpu/ollama-demo/__main__.py index 3945548..f26f7f3 100644 --- a/serverless/cloud-run-gpu/ollama-demo/__main__.py +++ b/serverless/cloud-run-gpu/ollama-demo/__main__.py @@ -40,6 +40,10 @@ context=docker_build.BuildContextArgs( location="./", ), + platforms=[ + docker_build.Platform.LINUX_AMD64, + docker_build.Platform.LINUX_ARM64, + ], push=True, )