ai-edge-torch-nightly 0.6.0.dev20250819__py3-none-any.whl → 0.6.0.dev20250820__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.
Files changed (7) hide show
  1. ai_edge_torch/generative/examples/deepseek/convert_to_tflite.py +3 -1
  2. ai_edge_torch/version.py +1 -1
  3. {ai_edge_torch_nightly-0.6.0.dev20250819.dist-info → ai_edge_torch_nightly-0.6.0.dev20250820.dist-info}/METADATA +1 -1
  4. {ai_edge_torch_nightly-0.6.0.dev20250819.dist-info → ai_edge_torch_nightly-0.6.0.dev20250820.dist-info}/RECORD +7 -7
  5. {ai_edge_torch_nightly-0.6.0.dev20250819.dist-info → ai_edge_torch_nightly-0.6.0.dev20250820.dist-info}/LICENSE +0 -0
  6. {ai_edge_torch_nightly-0.6.0.dev20250819.dist-info → ai_edge_torch_nightly-0.6.0.dev20250820.dist-info}/WHEEL +0 -0
  7. {ai_edge_torch_nightly-0.6.0.dev20250819.dist-info → ai_edge_torch_nightly-0.6.0.dev20250820.dist-info}/top_level.txt +0 -0
ai_edge_torch/generative/examples/deepseek/convert_to_tflite.py CHANGED
@@ -21,7 +21,9 @@ from ai_edge_torch.generative.utilities import converter
21
21
  from ai_edge_torch.generative.utilities import export_config
22
22
  from ai_edge_torch.generative.utilities import loader
23
23
 
24
- flags = converter.define_conversion_flags('deepseek')
24
+ flags = converter.define_conversion_flags(
25
+ 'deepseek', default_mask_as_input=True, default_transpose_kv_cache=True
26
+ )
25
27
 
26
28
 
27
29
  def main(_):
ai_edge_torch/version.py CHANGED
@@ -15,4 +15,4 @@
15
15
 
16
16
  # The next version of ai-edge-torch.
17
17
  # The minor version code should be bumped after every release.
18
- __version__ = "0.6.0.dev20250819"
18
+ __version__ = "0.6.0.dev20250820"
{ai_edge_torch_nightly-0.6.0.dev20250819.dist-info → ai_edge_torch_nightly-0.6.0.dev20250820.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: ai-edge-torch-nightly
3
- Version: 0.6.0.dev20250819
3
+ Version: 0.6.0.dev20250820
4
4
  Summary: Supporting PyTorch models with the Google AI Edge TFLite runtime.
5
5
  Home-page: https://github.com/google-ai-edge/ai-edge-torch
6
6
  Keywords: On-Device ML,AI,Google,TFLite,PyTorch,LLMs,GenAI
{ai_edge_torch_nightly-0.6.0.dev20250819.dist-info → ai_edge_torch_nightly-0.6.0.dev20250820.dist-info}/RECORD RENAMED
@@ -2,7 +2,7 @@ ai_edge_torch/__init__.py,sha256=lemyLCNoGYRnJsmDuGZu7qOqLbLqG6CGDFtu3ue1syU,129
2
2
  ai_edge_torch/_config.py,sha256=AiqhbcheF7j_ozIGDLC89k1we95aVgFDa-tR6h7UI0s,2529
3
3
  ai_edge_torch/conftest.py,sha256=r0GTrhMRhlmOGrrkvumHN8hkmyug6WvF60vWq8wRIBI,758
4
4
  ai_edge_torch/model.py,sha256=A7loFu8jE9CsXsfMmHYZ-KDFJiaD8Kkqwm_9d3IVzk0,5638
5
- ai_edge_torch/version.py,sha256=uaO5vWiEYZ6MPWlyKgPAo558dZdsAZEE69sX0BrLU5s,806
5
+ ai_edge_torch/version.py,sha256=V4nOyamqILX5YIedVKX9KourLPiAlWU8uLpDtvXf238,806
6
6
  ai_edge_torch/_convert/__init__.py,sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g,671
7
7
  ai_edge_torch/_convert/conversion.py,sha256=iQk3R-pLq4c1nfLqPB4xTRj78gghxPGzJCJtILLdg5o,6123
8
8
  ai_edge_torch/_convert/conversion_utils.py,sha256=Sr8qXVcTwc-ZnZmK7yxVrIOOp1S_vNrwzC0zUvLTI2o,2160
@@ -58,7 +58,7 @@ ai_edge_torch/generative/examples/amd_llama_135m/convert_to_tflite.py,sha256=ze0
58
58
  ai_edge_torch/generative/examples/amd_llama_135m/verify.py,sha256=uyBg5-trxQEjEHDZMX4qojkcsZgERUiPqIgR9n0_AY4,1311
59
59
  ai_edge_torch/generative/examples/amd_llama_135m/verify_util.py,sha256=J_FwH5KzS1tlyxcICSPir35gP2tWqRQ5wyVTsV9Uk8w,3018
60
60
  ai_edge_torch/generative/examples/deepseek/__init__.py,sha256=JaAnrFoXTl3RJX97XspklkTyqOHVyAgRJsZtzNDd10c,671
61
- ai_edge_torch/generative/examples/deepseek/convert_to_tflite.py,sha256=DvoHDBNc0GPW3kOv0TbGg0PnNTKijW_6z9LfQMku3RY,1816
61
+ ai_edge_torch/generative/examples/deepseek/convert_to_tflite.py,sha256=wcAk6EgNWXzYx_vQCmMDAAYw398ztlUdk_Gj9CYTsOg,1883
62
62
  ai_edge_torch/generative/examples/deepseek/deepseek.py,sha256=cm3rLw6aHFWQswwCqfvwxr05JG0maSzNVn1pBqD19qY,2816
63
63
  ai_edge_torch/generative/examples/deepseek/verify.py,sha256=HkvgEyGb-V_f6mWfyeN7Ai5uADAVQNzWvkygaKJiLAc,1344
64
64
  ai_edge_torch/generative/examples/deepseek/verify_util.py,sha256=xf2-jhv-ORwr_-Rs2UBpO2vcAwJNZIdmOxpcuKIQaww,3028
@@ -269,8 +269,8 @@ ai_edge_torch/testing/__init__.py,sha256=_yGgvnBZWb7T3IN3mc4x1sS4vM96HZwM8pwIcPG
269
269
  ai_edge_torch/testing/export.py,sha256=k5mGDGzwc23Z4zaIVDs8CNh-oOt64gsf9MS9NjhbPy4,3293
270
270
  ai_edge_torch/testing/model_coverage/__init__.py,sha256=5P8J6Zk5YYtDvTBucFvB9NGSRI7Gw_24WnrbhXgycEE,765
271
271
  ai_edge_torch/testing/model_coverage/model_coverage.py,sha256=UPB448aMDUyC0HNYVqio2rcJPnDN0tBQMP08J6vPYew,4718
272
- ai_edge_torch_nightly-0.6.0.dev20250819.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
273
- ai_edge_torch_nightly-0.6.0.dev20250819.dist-info/METADATA,sha256=idGFpPoSTWqEMO3SV4R9HfOxRmTDfJ_RgcmYBvm70T4,2074
274
- ai_edge_torch_nightly-0.6.0.dev20250819.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
275
- ai_edge_torch_nightly-0.6.0.dev20250819.dist-info/top_level.txt,sha256=5KXRaF2hwkApYxf7Y8y_tVb9aulGTlbOoNdbx1aKRkE,14
276
- ai_edge_torch_nightly-0.6.0.dev20250819.dist-info/RECORD,,
272
+ ai_edge_torch_nightly-0.6.0.dev20250820.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
273
+ ai_edge_torch_nightly-0.6.0.dev20250820.dist-info/METADATA,sha256=XYj1_J1mQgyoE6H2gPERRjUrt1oOpEvfKpCeZ13AL28,2074
274
+ ai_edge_torch_nightly-0.6.0.dev20250820.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
275
+ ai_edge_torch_nightly-0.6.0.dev20250820.dist-info/top_level.txt,sha256=5KXRaF2hwkApYxf7Y8y_tVb9aulGTlbOoNdbx1aKRkE,14
276
+ ai_edge_torch_nightly-0.6.0.dev20250820.dist-info/RECORD,,