diff --git a/shark/iree_utils/compile_utils.py b/shark/iree_utils/compile_utils.py
index 918f4dfc3a..2a7b6dba8a 100644
--- a/shark/iree_utils/compile_utils.py
+++ b/shark/iree_utils/compile_utils.py
@@ -95,6 +95,7 @@ def get_iree_frontend_args(frontend):
 def get_iree_common_args():
     return [
         "--iree-stream-resource-index-bits=64",
+        "--iree-stream-resource-max-allocation-size=4294967295",
         "--iree-vm-target-index-bits=64",
         "--iree-vm-bytecode-module-strip-source-map=true",
         "--iree-util-zero-fill-elided-attrs",