Further reduce memory usage in Quantizer

PiperOrigin-RevId: 679627364
google-ai-edge · Oct 4, 2024 · 082a2fa · 082a2fa
1 parent 66c56f0
commit 082a2fa
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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