ai-edge-quantizer-nightly 0.4.0.dev20250902__py3-none-any.whl → 0.4.0.dev20250904__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -14,6 +14,7 @@
14
14
  # ==============================================================================
15
15
 
16
16
  import os
17
+ import unittest # pylint: disable=unused-import, required for OSS.
17
18
 
18
19
  from absl.testing import parameterized
19
20
 
@@ -57,6 +58,7 @@ class RecipeTest(parameterized.TestCase):
57
58
  self.assertIsNotNone(quantization_result.quantized_model)
58
59
  return quantization_result
59
60
 
61
+ @unittest.skip('skipping due to b/438971945')
60
62
  def test_quantization_from_dynamic_wi8_afp32_func_succeeds(self):
61
63
  quant_result = self._quantize_with_recipe_func(
62
64
  recipe.dynamic_wi8_afp32, self._test_model_path
@@ -66,6 +68,7 @@ class RecipeTest(parameterized.TestCase):
66
68
  os.path.getsize(self._test_model_path),
67
69
  )
68
70
 
71
+ @unittest.skip('skipping due to b/438971945')
69
72
  def test_quantization_from_dynamic_wi4_afp32_func_succeeds(self):
70
73
  quant_result = self._quantize_with_recipe_func(
71
74
  recipe.dynamic_wi4_afp32, self._test_model_path
@@ -75,6 +78,7 @@ class RecipeTest(parameterized.TestCase):
75
78
  os.path.getsize(self._test_model_path),
76
79
  )
77
80
 
81
+ @unittest.skip('skipping due to b/438971945')
78
82
  def test_quantization_from_weight_only_wi8_afp32_func_succeeds(self):
79
83
  quant_result = self._quantize_with_recipe_func(
80
84
  recipe.weight_only_wi8_afp32, self._test_model_path
@@ -84,6 +88,7 @@ class RecipeTest(parameterized.TestCase):
84
88
  os.path.getsize(self._test_model_path),
85
89
  )
86
90
 
91
+ @unittest.skip('skipping due to b/438971945')
87
92
  def test_quantization_from_weight_only_wi4_afp32_func_succeeds(self):
88
93
  quant_result = self._quantize_with_recipe_func(
89
94
  recipe.weight_only_wi4_afp32, self._test_model_path
@@ -135,6 +140,7 @@ class RecipeTest(parameterized.TestCase):
135
140
  recipe_func=recipe.static_wi8_ai16,
136
141
  ),
137
142
  )
143
+ @unittest.skip('skipping due to b/438971945')
138
144
  def test_recipe_func_and_json_matches(self, recipe_json_path, recipe_func):
139
145
  # Quantize with recipe from function in recipe module.
140
146
  quant_result_from_func = self._quantize_with_recipe_func(
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: ai-edge-quantizer-nightly
3
- Version: 0.4.0.dev20250902
3
+ Version: 0.4.0.dev20250904
4
4
  Summary: A quantizer for advanced developers to quantize converted AI Edge models.
5
5
  Home-page: https://github.com/google-ai-edge/ai-edge-quantizer
6
6
  Keywords: On-Device ML,AI,Google,TFLite,Quantization,LLMs,GenAI
@@ -26,8 +26,8 @@ Description-Content-Type: text/markdown
26
26
  License-File: LICENSE
27
27
  Requires-Dist: immutabledict
28
28
  Requires-Dist: numpy
29
- Requires-Dist: ai-edge-tensorflow==2.21.0.dev20250818
30
- Requires-Dist: ai-edge-litert==1.4.*
29
+ Requires-Dist: tf-nightly
30
+ Requires-Dist: ai-edge-litert-nightly
31
31
 
32
32
  It aims to facilitate advanced users to strive for optimal performance on
33
33
  resource demanding models (e.g., GenAI models).
@@ -18,7 +18,7 @@ ai_edge_quantizer/quantizer_test.py,sha256=m6f4ayyaF3yQb9i4V0aFAbmGw0OKZ2Zam1RoT
18
18
  ai_edge_quantizer/recipe.py,sha256=MEkfQ2Sg3KAE9LAORHWcbjYNPg06EUbwc1d-VspQA2U,6461
19
19
  ai_edge_quantizer/recipe_manager.py,sha256=6dgbE-IZfEetzXH3p3Qm_9eQutNDOpZnMpiaLTbP-ZQ,14744
20
20
  ai_edge_quantizer/recipe_manager_test.py,sha256=H-B75vwPN5ND-nUa3pOXizeHTv4mufPiC5cL_OlDIYU,34040
21
- ai_edge_quantizer/recipe_test.py,sha256=GKuo6N65wKLS2xwSpjd-BWWeVRpF1zc7Yt7phSMYSxA,5905
21
+ ai_edge_quantizer/recipe_test.py,sha256=QisyaTol8JRZFcGOGyee7QRCvqj5VbF4guKWdIoMUOE,6213
22
22
  ai_edge_quantizer/transformation_instruction_generator.py,sha256=O0U2aZcB8aXQgOV8r9g1rGNzDUiuI5Ta53XnxZbVffE,31576
23
23
  ai_edge_quantizer/transformation_instruction_generator_test.py,sha256=KW5-WoTTo9IqLEVnWxVC8ut8eWLi_91xfKgGqVQ9QDk,54635
24
24
  ai_edge_quantizer/transformation_performer.py,sha256=o4J6OUbI0dLoobVYjkOFw5Po3yH0gZJXrfuTIYais4o,13029
@@ -70,8 +70,8 @@ ai_edge_quantizer/utils/tfl_interpreter_utils.py,sha256=EoVjI_hplX_Rml3hfRsGmQOi
70
70
  ai_edge_quantizer/utils/tfl_interpreter_utils_test.py,sha256=6fjkM-rycZ95L4yfvlr0TN6RlrhfPzxNUYrZaYO_F0A,12013
71
71
  ai_edge_quantizer/utils/validation_utils.py,sha256=oYw33Sg547AqtGw-choPUJmp9SAKkV46J_ddqSsum2Q,3950
72
72
  ai_edge_quantizer/utils/validation_utils_test.py,sha256=V_qNDikPD4OPB-siOLQCWNVWTAu87h2IgNYt7teFd-o,2934
73
- ai_edge_quantizer_nightly-0.4.0.dev20250902.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
74
- ai_edge_quantizer_nightly-0.4.0.dev20250902.dist-info/METADATA,sha256=t3d33pChyc9pXD3vZDxORVTxU04T6Kuen4PMcCc4sdk,1535
75
- ai_edge_quantizer_nightly-0.4.0.dev20250902.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
76
- ai_edge_quantizer_nightly-0.4.0.dev20250902.dist-info/top_level.txt,sha256=8QTfPnFXNVUhScFLaa-NWZMFWMn72M50DVPubpwWB1g,18
77
- ai_edge_quantizer_nightly-0.4.0.dev20250902.dist-info/RECORD,,
73
+ ai_edge_quantizer_nightly-0.4.0.dev20250904.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
74
+ ai_edge_quantizer_nightly-0.4.0.dev20250904.dist-info/METADATA,sha256=AqVFUY7NzeZmDKSjDGACfkR1HIsitVwnnJ91NlaktX0,1508
75
+ ai_edge_quantizer_nightly-0.4.0.dev20250904.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
76
+ ai_edge_quantizer_nightly-0.4.0.dev20250904.dist-info/top_level.txt,sha256=8QTfPnFXNVUhScFLaa-NWZMFWMn72M50DVPubpwWB1g,18
77
+ ai_edge_quantizer_nightly-0.4.0.dev20250904.dist-info/RECORD,,