ai-edge-torch-nightly 0.5.0.dev20250506__py3-none-any.whl → 0.5.0.dev20250507__py3-none-any.whl

ai_edge_torch/generative/examples/gemma3/decoder.py

@@ -201,7 +201,7 @@ class Decoder(nn.Module):
         torch.zeros_like(sliding_mask_bool, dtype=torch.float),
         torch.full_like(
             sliding_mask_bool,
-            self.config.get_causal_mask_value(),
+            self.config.causal_mask_value,
             dtype=torch.float,
         ),
     )
@@ -219,7 +219,7 @@ class Decoder(nn.Module):
       mask = torch.logical_and(mask, pixel_mask)
     else:
       mask = torch.logical_or(mask, pixel_mask)
-    mask = torch.where(mask, 0, self.config.get_causal_mask_value())
+    mask = torch.where(mask, 0, self.config.causal_mask_value)
     return mask
 
   def build_pixel_mask(self, image_indices: torch.Tensor):

ai_edge_torch/generative/layers/model_config.py

@@ -251,5 +251,5 @@ class ModelConfig:
     return self.block_configs[idx]
 
   @property
-  def get_causal_mask_value(self) -> float:
+  def causal_mask_value(self) -> float:
     return self.block_config(0).attn_config.causal_mask_value

ai_edge_torch/generative/quantize/quant_attrs.py

@@ -23,6 +23,7 @@ class Dtype(enum.Enum):
   FP32 = enum.auto()
   FP16 = enum.auto()
   INT8 = enum.auto()
+  INT4 = enum.auto()
 
 
 @enum.unique
@@ -66,3 +67,4 @@ class Granularity(enum.Enum):
 
   NONE = enum.auto()
   CHANNELWISE = enum.auto()
+  BLOCKWISE = enum.auto()

ai_edge_torch/generative/quantize/quant_recipe.py

@@ -36,6 +36,7 @@ class LayerQuantRecipe:
     mode: Type of quantization.
     algorithm: Algorithm for calculating quantization parameters.
     granularity: Granularity of quantization.
+    block_size: Size of the block for blockwise quantization.
   """
 
   activation_dtype: quant_attrs.Dtype
@@ -43,15 +44,18 @@ class LayerQuantRecipe:
   mode: quant_attrs.Mode
   algorithm: quant_attrs.Algorithm
   granularity: quant_attrs.Granularity
+  block_size: int = 0
 
   def __str__(self):
-    return (
+    base_str = (
         f'(a:{self.activation_dtype.name}, '
         f'w:{self.weight_dtype.name}, '
         f'{self.mode.name}, '
         f'{self.algorithm.name}, '
-        f'{self.granularity.name})'
+        f'{self.granularity.name}'
+        f'{self.block_size}'
     )
+    return f'{base_str})'
 
   __repr__ = __str__
 
@@ -70,6 +74,16 @@ class LayerQuantRecipe:
           and self.algorithm == supported[3]
           and self.granularity == supported[4]
       ):
+        if self.block_size > 0:
+          if (
+              self.block_size % 32 == 0
+              and self.granularity == quant_attrs.Granularity.BLOCKWISE
+          ):
+            is_valid = True
+            break
+          else:
+            is_valid = False
+            break
         is_valid = True
         break
 

ai_edge_torch/generative/quantize/quant_recipe_utils.py

@@ -60,3 +60,16 @@ def create_layer_quant_fp16() -> quant_recipe.LayerQuantRecipe:
       algorithm=quant_attrs.Algorithm.FLOAT_CAST,
       granularity=quant_attrs.Granularity.NONE,
   )
+
+
+def create_layer_quant_int4_dynamic_block(
+    block_size: int,
+) -> quant_recipe.LayerQuantRecipe:
+  return quant_recipe.LayerQuantRecipe(
+      activation_dtype=quant_attrs.Dtype.FP32,
+      weight_dtype=quant_attrs.Dtype.INT4,
+      mode=quant_attrs.Mode.DYNAMIC_RANGE,
+      algorithm=quant_attrs.Algorithm.MIN_MAX,
+      granularity=quant_attrs.Granularity.BLOCKWISE,
+      block_size=block_size,
+  )

ai_edge_torch/generative/quantize/quant_recipes.py

@@ -54,3 +54,15 @@ def full_fp16_recipe() -> quant_config.QuantConfig:
           default=quant_recipe_utils.create_layer_quant_fp16()
       )
   )
+
+
+def all_supported_int4_dynamic_block_recipe(
+    block_size: int,
+) -> quant_config.QuantConfig:
+  return quant_config.QuantConfig(
+      generative_recipe=quant_recipe.GenerativeQuantRecipe(
+          default=quant_recipe_utils.create_layer_quant_int4_dynamic_block(
+              block_size
+          )
+      )
+  )

ai_edge_torch/generative/quantize/supported_schemes.py

@@ -29,4 +29,5 @@ def get_supported_layer_schemes():
       (_t.FP32, _t.INT8, _m.DYNAMIC_RANGE, _a.MIN_MAX, _g.CHANNELWISE),
       (_t.FP32, _t.INT8, _m.WEIGHT_ONLY, _a.MIN_MAX, _g.CHANNELWISE),
       (_t.FP32, _t.FP16, _m.WEIGHT_ONLY, _a.FLOAT_CAST, _g.NONE),
+      (_t.FP32, _t.INT4, _m.DYNAMIC_RANGE, _a.MIN_MAX, _g.BLOCKWISE),
   ]

ai_edge_torch/generative/test/test_quantize.py

@@ -40,6 +40,7 @@ class TestVerifyRecipes(parameterized.TestCase):
       (Dtype.INT8, Dtype.FP16),
       (Dtype.FP16, Dtype.INT8),
       (Dtype.FP16, Dtype.FP16),
+      (Dtype.FP16, Dtype.INT4),
   ])
   def test_verify_invalid_recipes(
       self,
@@ -74,6 +75,14 @@ class TestVerifyRecipes(parameterized.TestCase):
           Algorithm.FLOAT_CAST,
           Granularity.NONE,
       ),
+      (
+          Dtype.FP32,
+          Dtype.INT4,
+          Mode.DYNAMIC_RANGE,
+          Algorithm.MIN_MAX,
+          Granularity.BLOCKWISE,
+          32,
+      ),
   ])
   def test_verify_valid_recipes(
       self,
@@ -82,6 +91,7 @@ class TestVerifyRecipes(parameterized.TestCase):
       mode,
       algo,
       granularity,
+      block_size=None,
   ):
     quant_recipe.LayerQuantRecipe(
         activation, weight, mode, algo, granularity
@@ -150,6 +160,33 @@ class TestQuantizeConvert(parameterized.TestCase):
         "Quantized model isn't smaller than F32 model.",
     )
 
+  def test_quantize_convert_toy_blockwise(self):
+    config = toy_model.get_model_config()
+    pytorch_model = toy_model.ToySingleLayerModel(config)
+    idx = torch.unsqueeze(torch.arange(0, 100, dtype=torch.int), 0)
+    input_pos = torch.arange(0, 100, dtype=torch.int)
+    quant_config = quant_recipes.all_supported_int4_dynamic_block_recipe(32)
+    quantized_model = ai_edge_torch.convert(
+        pytorch_model, (idx, input_pos), quant_config=quant_config
+    )
+    float_model = ai_edge_torch.convert(pytorch_model, (idx, input_pos))
+    self.assertLess(
+        len(quantized_model._tflite_model),
+        len(float_model._tflite_model),
+        "Quantized model isn't smaller than F32 model.",
+    )
+
+  def test_unsupported_block_size(self):
+    config = toy_model.get_model_config()
+    pytorch_model = toy_model.ToySingleLayerModel(config)
+    idx = torch.unsqueeze(torch.arange(0, 100, dtype=torch.int), 0)
+    input_pos = torch.arange(0, 100, dtype=torch.int)
+    self.assertRaises(
+        ValueError,
+        quant_recipes.all_supported_int4_dynamic_block_recipe,
+        36,
+    )
+
   def test_quantize_convert_compare_toy(self):
     self.skipTest("b/338288901")
     config = toy_model_with_kv_cache.get_model_config()

ai_edge_torch/generative/utilities/converter.py

@@ -243,15 +243,13 @@ def _export_helper(
 
   prefill_masks = None
   if flags.FLAGS.mask_as_input:
-    prefill_masks = [
-        _build_mask(
-            flags.FLAGS.prefill_seq_lens,
-            flags.FLAGS.kv_cache_max_len,
-            config.get_causal_mask_value(),
-        )
-    ]
-
-  if prefill_masks:
+    prefill_masks = _build_mask(
+        flags.FLAGS.prefill_seq_lens,
+        flags.FLAGS.kv_cache_max_len,
+        config.causal_mask_value,
+    )
+    if not isinstance(prefill_masks, list):
+      prefill_masks = [prefill_masks]
     assert len(prefill_masks) == len(prefill_seq_lens)
 
   decode_token = torch.tensor(
@@ -321,7 +319,7 @@ def _export_helper(
       #  torch.triu(mask, diagonal=decode_position).unsqueeze(0).unsqueeze(0)
       #
       sample_kwargs['mask'] = _build_mask(
-          1, flags.FLAGS.kv_cache_max_len, config.get_causal_mask_value()
+          1, flags.FLAGS.kv_cache_max_len, config.causal_mask_value
       )
     if lora is not None:
       sample_kwargs['lora'] = lora

ai_edge_torch/lowertools/translate_recipe.py

@@ -13,9 +13,9 @@
 # limitations under the License.
 # ==============================================================================
 
-from ai_edge_quantizer import quantizer
 from ai_edge_torch.generative.quantize import quant_attrs
 from ai_edge_torch.generative.quantize import quant_recipe
+from ai_edge_quantizer import quantizer
 
 _ComputePrecision = quantizer.qtyping.ComputePrecision
 _QuantGranularity = quantizer.qtyping.QuantGranularity
@@ -39,6 +39,8 @@ def _get_nbits_from_dtype(dtype: quant_attrs.Dtype) -> int:
     return 16
   elif dtype == quant_attrs.Dtype.INT8:
     return 8
+  elif dtype == quant_attrs.Dtype.INT4:
+    return 4
   raise ValueError('Unimplemented number of bits')
 
 
@@ -76,6 +78,8 @@ def _get_granularity(
     return _QuantGranularity.CHANNELWISE
   if granularity == quant_attrs.Granularity.NONE:
     return _QuantGranularity.TENSORWISE
+  if granularity == quant_attrs.Granularity.BLOCKWISE:
+    return _QuantGranularity.BLOCKWISE
   raise ValueError('Unimplemented granularity')
 
 
@@ -101,6 +105,7 @@ def _set_quant_config(
               symmetric=True,
               granularity=_get_granularity(layer_recipe.granularity),
               dtype=_get_dtype_from_dtype(layer_recipe.weight_dtype),
+              block_size=layer_recipe.block_size,
           ),
           compute_precision=_get_compute_precision_from_mode(layer_recipe.mode),
           explicit_dequantize=_get_explicit_dequant_from_mode(

ai_edge_torch/version.py

@@ -13,4 +13,4 @@
 # limitations under the License.
 # ==============================================================================
 
-__version__ = "0.5.0.dev20250506"
+__version__ = "0.5.0.dev20250507"

{ai_edge_torch_nightly-0.5.0.dev20250506.dist-info → ai_edge_torch_nightly-0.5.0.dev20250507.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ai-edge-torch-nightly
-Version: 0.5.0.dev20250506
+Version: 0.5.0.dev20250507
 Summary: Supporting PyTorch models with the Google AI Edge TFLite runtime.
 Home-page: https://github.com/google-ai-edge/ai-edge-torch
 Keywords: On-Device ML,AI,Google,TFLite,PyTorch,LLMs,GenAI

{ai_edge_torch_nightly-0.5.0.dev20250506.dist-info → ai_edge_torch_nightly-0.5.0.dev20250507.dist-info}/RECORD

@@ -2,7 +2,7 @@ ai_edge_torch/__init__.py,sha256=8sPR_5uXJA4NEE0nIwNdSl-ADOJEoR8hAgYvBQDY70Y,120
 ai_edge_torch/_config.py,sha256=AiqhbcheF7j_ozIGDLC89k1we95aVgFDa-tR6h7UI0s,2529
 ai_edge_torch/conftest.py,sha256=r0GTrhMRhlmOGrrkvumHN8hkmyug6WvF60vWq8wRIBI,758
 ai_edge_torch/model.py,sha256=wxjSFq_rBSxSqbUE8E8EJTCkgvgaRLjq_ZuAM-IZpCU,5606
-ai_edge_torch/version.py,sha256=h2DvepK-jWas3mxk9s_2MbivYh_uy9-TIfaXJN7FlTY,706
+ai_edge_torch/version.py,sha256=87Gl0LNfMSQT4gAxzJTQ7v7ZnE7ZMa1QhSbSsKLUIW8,706
 ai_edge_torch/_convert/__init__.py,sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g,671
 ai_edge_torch/_convert/conversion.py,sha256=QVugYVfbyaeBgSKKbhFzHG5oXA7t3M-40JcpcdSu6W8,5436
 ai_edge_torch/_convert/conversion_utils.py,sha256=Sr8qXVcTwc-ZnZmK7yxVrIOOp1S_vNrwzC0zUvLTI2o,2160
@@ -69,7 +69,7 @@ ai_edge_torch/generative/examples/gemma/verify_gemma2.py,sha256=jhiyinOqPt5ZZjEa
 ai_edge_torch/generative/examples/gemma/verify_util.py,sha256=n7f2nF6Lin_tDvPs0JVldsuaBzo7pAwi5YAHAhlIxQg,6139
 ai_edge_torch/generative/examples/gemma3/__init__.py,sha256=JaAnrFoXTl3RJX97XspklkTyqOHVyAgRJsZtzNDd10c,671
 ai_edge_torch/generative/examples/gemma3/convert_gemma3_to_tflite.py,sha256=MjkQDVynaw9C5z9ODzKfb85xW5JfxHUWBJ_Aco05FHo,1760
-ai_edge_torch/generative/examples/gemma3/decoder.py,sha256=fzLpuJO5JseQLA38Li-i9Xdnh9I4zdBWQEOeNbUEfjI,15737
+ai_edge_torch/generative/examples/gemma3/decoder.py,sha256=xGxeNKQvgyrENmUQMu0uKymL3qthvbdoxdMbAzwiLz0,15725
 ai_edge_torch/generative/examples/gemma3/gemma3.py,sha256=GACDBI_MsFowR8A3wAWrpzradPYe-AUgB9ZjXaVBG-s,6485
 ai_edge_torch/generative/examples/gemma3/image_encoder.py,sha256=uRoLoBWzFtQz5wFZfPCxbkvZsgPAqSkUUsV3977GbYc,5184
 ai_edge_torch/generative/examples/gemma3/verify_gemma3.py,sha256=v8oNXFICmVOtQxfO7IhZ8GnbvotEkDi9lzYHjoQyOso,2464
@@ -165,7 +165,7 @@ ai_edge_torch/generative/layers/feed_forward.py,sha256=hdICat-8gW7-vxDAevJQ8NQ-m
 ai_edge_torch/generative/layers/feed_forward_test.py,sha256=8ZGy79BBpsyS6yKKDEKrDt249G5Mz-8VKWW7_WHx0u4,1655
 ai_edge_torch/generative/layers/kv_cache.py,sha256=b-7shzDaKexmvQF7P3SiAmIz4ZofjYWv3m5u71GojsA,10460
 ai_edge_torch/generative/layers/lora.py,sha256=hsvWLLOnW7HQ0AysOZu30x_cetMquDd1tjfyLz8HCSU,17892
-ai_edge_torch/generative/layers/model_config.py,sha256=dRZUMa71ADaEllu7TfXUWTMHRCcMgvkFMYMzmeJi4G8,8576
+ai_edge_torch/generative/layers/model_config.py,sha256=X_gjN5524DCDBNXsX5GrOBlkKM4UHzj_RfdCD0-VOxQ,8572
 ai_edge_torch/generative/layers/normalization.py,sha256=MbwH-n80Fob5YvjBzdqDjBizMHLzSJGYRDdbD-rL5C0,6174
 ai_edge_torch/generative/layers/rotary_position_embedding.py,sha256=975zR202MdIrILJ7blceAcxrNqX1ZCN0ECKG1gz-bV8,2655
 ai_edge_torch/generative/layers/scaled_dot_product_attention.py,sha256=2_AgwENsaOgaxgiSqgoj0V0JzQ09dFtP_nBhX-lJK2g,5648
@@ -177,11 +177,11 @@ ai_edge_torch/generative/layers/unet/builder.py,sha256=zAqWXdimmMrQRhmE_t9XkS68m
 ai_edge_torch/generative/layers/unet/model_config.py,sha256=pPDwLawc23pfMaPVyMJlYmxVVusjMvx-l8wBwOYOH-c,9692
 ai_edge_torch/generative/quantize/__init__.py,sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g,671
 ai_edge_torch/generative/quantize/example.py,sha256=1lfVNUd2cEyRUnoZ7BLbRJ9IN-FTKiWBtZNPFUzAiWE,1747
-ai_edge_torch/generative/quantize/quant_attrs.py,sha256=n1Fm8BFC8gJa_oiwwAOOghJyHtOXYZ4q-5ZRy4pHrIw,1957
-ai_edge_torch/generative/quantize/quant_recipe.py,sha256=tKnuJq6hPD23JPCB9nPAlE1UHAwdbChkgPShiVaz4CE,5156
-ai_edge_torch/generative/quantize/quant_recipe_utils.py,sha256=4fgmP_GgeiFUOkIaC9ZZXC12eO3DQZdrWDXRz5YXiwU,2270
-ai_edge_torch/generative/quantize/quant_recipes.py,sha256=0Kvr_o7pbMnE8VMe6Ml0FBxkHM6RJ3C14B2I1mjItjc,2030
-ai_edge_torch/generative/quantize/supported_schemes.py,sha256=FjdycEOvxRgBmQdZVufetPvkDoD7rUowIOSKV9oV5Kk,1418
+ai_edge_torch/generative/quantize/quant_attrs.py,sha256=plMsd7JBi98r2NHsAdMdvS6TPTXAoRFLCwOXu8H3-24,2004
+ai_edge_torch/generative/quantize/quant_recipe.py,sha256=3xT4N5tfggXJqgwKW4ntIkwsrNVtkG2SIUHeiSF5yOs,5579
+ai_edge_torch/generative/quantize/quant_recipe_utils.py,sha256=h3k_na6rbR08Ip79-2JbkeH8RDk_rrnEGiytuzFDhqc,2678
+ai_edge_torch/generative/quantize/quant_recipes.py,sha256=a71KFHVbjJdBDpYshbUI69NxGNOmPuqp_NZvNSrf00c,2349
+ai_edge_torch/generative/quantize/supported_schemes.py,sha256=TwR2FpQuBEORy6FshEyHNBMKARWlA2MVtTfX9tXV5aE,1488
 ai_edge_torch/generative/test/__init__.py,sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g,671
 ai_edge_torch/generative/test/test_custom_dus.py,sha256=MjIhTvkTko872M35XMciobvICcDWTcIDJ3rociko-wM,3267
 ai_edge_torch/generative/test/test_kv_cache.py,sha256=1sXN2RPntq0PP3IEy0NkvIbzQ0Y8JhPIwRSFwO9JLlE,5728
@@ -189,10 +189,10 @@ ai_edge_torch/generative/test/test_loader.py,sha256=9mQUeeZKOVApOWSWl2cN9c10axZj
 ai_edge_torch/generative/test/test_lora.py,sha256=6QIM6RLTc2HrodGpp_aS3OxM9Rco2KAzEnYgotkg41M,5310
 ai_edge_torch/generative/test/test_model_conversion.py,sha256=mhNJikLnGVGi9NKmXB8FhnqeDy9gtrvC3yEbrTABZ4Y,6163
 ai_edge_torch/generative/test/test_model_conversion_large.py,sha256=vQWmpzMkJ2hPmWpg41ZMWwBsngTykRVzRPHtpbkwiLM,12811
-ai_edge_torch/generative/test/test_quantize.py,sha256=bEJMhpQ9bIDUZVBXTW888728FcH-i3SyE4JSZZUgU0A,6071
+ai_edge_torch/generative/test/test_quantize.py,sha256=TG6vTF9yOZWe2wW7v8-hmuaQoODwJC1Z-2d5xv3zgfI,7389
 ai_edge_torch/generative/test/utils.py,sha256=tF6aCfAGJnc9dmzCnZCEOuKNVimfWOqscv9og0DDLHU,2656
 ai_edge_torch/generative/utilities/__init__.py,sha256=-_jxnnFnCgnTU4oTm4MnRsvL5lqhomBNdFBbqfmfHPo,720
-ai_edge_torch/generative/utilities/converter.py,sha256=K1gZWPq5f3Z7f9USeJ_PphctO1dyYTNrWSJQ-cztgKA,11658
+ai_edge_torch/generative/utilities/converter.py,sha256=d0JOWN5l2vbvt8RzFFiRoulkWiejyEZ21xKv5LdLIyc,11675
 ai_edge_torch/generative/utilities/export_config.py,sha256=5IvR3grlMd4mWO5c_Y4x9Fk1b1xa57MzlYNE8XUaN28,2049
 ai_edge_torch/generative/utilities/loader.py,sha256=7p__m2JryWphGlYOuRxdoT4id4_tWJEVOV7y2X4H-Ak,13737
 ai_edge_torch/generative/utilities/model_builder.py,sha256=ZYX1TxpFdj573du2QCyHJlFjx4q1m12R74fp4Gwl92A,6343
@@ -214,7 +214,7 @@ ai_edge_torch/lowertools/common_utils.py,sha256=4HQtquPZ6oiId8vR_1ykW_uK4ELnyo5z
 ai_edge_torch/lowertools/odml_torch_utils.py,sha256=QRuS7S5lULRWEh3J1sWIsnKh-rbX7rd9tt6JJHbMPfo,8317
 ai_edge_torch/lowertools/test_utils.py,sha256=mdxTlhqHABZEQ_GEmPFCL8LIAWtqRtYZUGdSY1ieZjw,1949
 ai_edge_torch/lowertools/torch_xla_utils.py,sha256=1EytIw2R6dthhLhf69wN1L9BaQTeybCD0wga-PhHcMI,9518
-ai_edge_torch/lowertools/translate_recipe.py,sha256=ymkBpFqAUiupRWqrPOWiVphKcXR1K5vHK0RjgBFtxlE,5652
+ai_edge_torch/lowertools/translate_recipe.py,sha256=kUVCe69_DzvfbNYVB0MY2rCZwWaN8t3NoNu8Vh4x5bQ,5849
 ai_edge_torch/odml_torch/__init__.py,sha256=S8jOzE9nLof-6es3XDiGJRN-9H_XTxsVm9dE7lD3RWo,812
 ai_edge_torch/odml_torch/_torch_future.py,sha256=jSYHf1CMTJzMizPMbu2b39hAt0ZTR6gQLq67GMe9KTo,2336
 ai_edge_torch/odml_torch/_torch_library.py,sha256=Lw1gqL2HWNRspdTwNhIkYAHDyafHedHtkXyKKxn-Wss,805
@@ -251,8 +251,8 @@ ai_edge_torch/testing/__init__.py,sha256=_yGgvnBZWb7T3IN3mc4x1sS4vM96HZwM8pwIcPG
 ai_edge_torch/testing/export.py,sha256=k5mGDGzwc23Z4zaIVDs8CNh-oOt64gsf9MS9NjhbPy4,3293
 ai_edge_torch/testing/model_coverage/__init__.py,sha256=5P8J6Zk5YYtDvTBucFvB9NGSRI7Gw_24WnrbhXgycEE,765
 ai_edge_torch/testing/model_coverage/model_coverage.py,sha256=UPB448aMDUyC0HNYVqio2rcJPnDN0tBQMP08J6vPYew,4718
-ai_edge_torch_nightly-0.5.0.dev20250506.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
-ai_edge_torch_nightly-0.5.0.dev20250506.dist-info/METADATA,sha256=U800OYIqsYFtAziBGVh7GJ__RvSvmkHAzcttf1Z3vME,2051
-ai_edge_torch_nightly-0.5.0.dev20250506.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-ai_edge_torch_nightly-0.5.0.dev20250506.dist-info/top_level.txt,sha256=5KXRaF2hwkApYxf7Y8y_tVb9aulGTlbOoNdbx1aKRkE,14
-ai_edge_torch_nightly-0.5.0.dev20250506.dist-info/RECORD,,
+ai_edge_torch_nightly-0.5.0.dev20250507.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
+ai_edge_torch_nightly-0.5.0.dev20250507.dist-info/METADATA,sha256=YKm-Nzrn7i9780PIOL6mO3vLxeCaGhJ_C4gSHTOthYI,2051
+ai_edge_torch_nightly-0.5.0.dev20250507.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+ai_edge_torch_nightly-0.5.0.dev20250507.dist-info/top_level.txt,sha256=5KXRaF2hwkApYxf7Y8y_tVb9aulGTlbOoNdbx1aKRkE,14
+ai_edge_torch_nightly-0.5.0.dev20250507.dist-info/RECORD,,