diff --git a/platform/src/components/aws/function.ts b/platform/src/components/aws/function.ts
index 40ce7cfcf..1ed5ac265 100644
--- a/platform/src/components/aws/function.ts
+++ b/platform/src/components/aws/function.ts
@@ -1620,6 +1620,12 @@ export class Function extends Component implements Link.Linkable {
               registryId: bootstrapData.assetEcrRegistryId,
             });
 
+            const archPlatformMap: Record<string, Platform> = {
+              "arm64": "linux/arm64",
+              "x86_64": "linux/amd64"
+            }
+
+            const platforms = architectures.apply((architectures) => architectures.map((arch) => archPlatformMap[arch]))
             // build image
             //aws-python-container::sst:aws:Function::MyPythonFunction
             return new Image(
@@ -1658,7 +1664,7 @@ export class Function extends Component implements Link.Linkable {
                     inline: {},
                   },
                 ],
-                /// TODO: walln - enable arm64 builds by using architecture args
+                platforms,
                 push: true,
                 registries: [
                   authToken.apply((authToken) => ({