increase allocated model architecture size to 64MB

PiperOrigin-RevId: 693390872
google-ai-edge · Nov 5, 2024 · 4a5d2ad · 4a5d2ad
1 parent 7de9eb7
commit 4a5d2ad
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/ai_edge_quantizer/model_modifier.py b/ai_edge_quantizer/model_modifier.py
@@ -70,7 +70,8 @@ def modify_model(
         instructions, quantized_model
     )
     constant_buffer_size = self._process_constant_map(quantized_model)
-    if constant_buffer_size > 2**31 - 2**20:
+    # we leave 64MB for the model architecture.
+    if constant_buffer_size > 2**31 - 2**26:
       return self._serialize_large_model(quantized_model)
     else:
       return self._serialize_small_model(quantized_model)