From ed0d0ccf7346754ca3fdc38cab11cb59f14d9ba9 Mon Sep 17 00:00:00 2001
From: Vitalii Dziuba <vdziuba@google.com>
Date: Fri, 27 Sep 2024 09:48:56 -0700
Subject: [PATCH] Further reduce memory usage in Quantizer

PiperOrigin-RevId: 679627364
---
 ai_edge_torch/lowertools/translate_recipe.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/ai_edge_torch/lowertools/translate_recipe.py b/ai_edge_torch/lowertools/translate_recipe.py
index e703d61..ce42dad 100644
--- a/ai_edge_torch/lowertools/translate_recipe.py
+++ b/ai_edge_torch/lowertools/translate_recipe.py
@@ -156,8 +156,8 @@ def translate_to_ai_edge_recipe(
 
 
 def quantize_model(
-    model: bytearray, recipe: quantizer.recipe_manager.ModelQuantizationRecipe
+    model: bytes, recipe: quantizer.recipe_manager.ModelQuantizationRecipe
 ) -> bytearray:
-  qt = quantizer.Quantizer(bytearray(model), recipe)
+  qt = quantizer.Quantizer(model, recipe)
   result = qt.quantize()
   return result.quantized_model