ai-edge-torch-nightly 0.7.0.dev20251031__py3-none-any.whl → 0.7.0.dev20251102__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.
- ai_edge_torch/generative/layers/model_config.py +1 -0
- ai_edge_torch/version.py +1 -1
- {ai_edge_torch_nightly-0.7.0.dev20251031.dist-info → ai_edge_torch_nightly-0.7.0.dev20251102.dist-info}/METADATA +13 -3
- {ai_edge_torch_nightly-0.7.0.dev20251031.dist-info → ai_edge_torch_nightly-0.7.0.dev20251102.dist-info}/RECORD +7 -7
- {ai_edge_torch_nightly-0.7.0.dev20251031.dist-info → ai_edge_torch_nightly-0.7.0.dev20251102.dist-info}/WHEEL +1 -1
- {ai_edge_torch_nightly-0.7.0.dev20251031.dist-info → ai_edge_torch_nightly-0.7.0.dev20251102.dist-info/licenses}/LICENSE +0 -0
- {ai_edge_torch_nightly-0.7.0.dev20251031.dist-info → ai_edge_torch_nightly-0.7.0.dev20251102.dist-info}/top_level.txt +0 -0
|
@@ -251,6 +251,7 @@ class ModelConfig:
|
|
|
251
251
|
lm_head_use_bias: bool = False
|
|
252
252
|
# Whether LLM's HEAD shares the weight of the embedding.
|
|
253
253
|
lm_head_share_weight_with_embedding: bool = True
|
|
254
|
+
dense_intermediate_size: Optional[int] = None
|
|
254
255
|
|
|
255
256
|
# Whether to turn on high-level function boundary.
|
|
256
257
|
enable_hlfb: bool = True
|
ai_edge_torch/version.py
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
|
-
Metadata-Version: 2.
|
|
1
|
+
Metadata-Version: 2.4
|
|
2
2
|
Name: ai-edge-torch-nightly
|
|
3
|
-
Version: 0.7.0.
|
|
3
|
+
Version: 0.7.0.dev20251102
|
|
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
|
|
@@ -39,7 +39,17 @@ Requires-Dist: ai-edge-quantizer-nightly
|
|
|
39
39
|
Requires-Dist: jax
|
|
40
40
|
Requires-Dist: torch-xla2[odml]>=0.0.1.dev20241201
|
|
41
41
|
Provides-Extra: torch-xla
|
|
42
|
-
Requires-Dist:
|
|
42
|
+
Requires-Dist: torch_xla>=2.4.0; extra == "torch-xla"
|
|
43
|
+
Dynamic: classifier
|
|
44
|
+
Dynamic: description
|
|
45
|
+
Dynamic: description-content-type
|
|
46
|
+
Dynamic: home-page
|
|
47
|
+
Dynamic: keywords
|
|
48
|
+
Dynamic: license-file
|
|
49
|
+
Dynamic: provides-extra
|
|
50
|
+
Dynamic: requires-dist
|
|
51
|
+
Dynamic: requires-python
|
|
52
|
+
Dynamic: summary
|
|
43
53
|
|
|
44
54
|
Library that supports converting PyTorch models into a .tflite format, which can
|
|
45
55
|
then be run with TensorFlow Lite and MediaPipe. This enables applications for
|
|
@@ -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=
|
|
5
|
+
ai_edge_torch/version.py,sha256=ozb0N4q30nIdG-3_G7H08LXrj4PjO6b9S1P5g_jcMHQ,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
|
|
@@ -180,7 +180,7 @@ ai_edge_torch/generative/layers/feed_forward.py,sha256=_GmtHxwL068l9gh_F_WFcFk7L
|
|
|
180
180
|
ai_edge_torch/generative/layers/feed_forward_test.py,sha256=Y5l1eC9NgfYixHcfIfE1W4FGh7oC-9UGGyHdKS9tQKc,1880
|
|
181
181
|
ai_edge_torch/generative/layers/kv_cache.py,sha256=A0IFXZ1HD2ZHOWRLfsDO4almgE0KQfjyBOdBFZIGnAs,10893
|
|
182
182
|
ai_edge_torch/generative/layers/lora.py,sha256=hsvWLLOnW7HQ0AysOZu30x_cetMquDd1tjfyLz8HCSU,17892
|
|
183
|
-
ai_edge_torch/generative/layers/model_config.py,sha256=
|
|
183
|
+
ai_edge_torch/generative/layers/model_config.py,sha256=y6nyydMzm7JUV7AKttcFy3tvti-nE6tRXoVbBB9dyiM,10438
|
|
184
184
|
ai_edge_torch/generative/layers/normalization.py,sha256=syasVh3dRDVp2Nwhl0x7zucL-chTnCqWgeV1mb87DFY,7435
|
|
185
185
|
ai_edge_torch/generative/layers/normalization_test.py,sha256=zwurZly-TgFxdgVVdpzu9vCpcLbd5RYt_gKg9Lfg1jI,2248
|
|
186
186
|
ai_edge_torch/generative/layers/rotary_position_embedding.py,sha256=975zR202MdIrILJ7blceAcxrNqX1ZCN0ECKG1gz-bV8,2655
|
|
@@ -270,8 +270,8 @@ ai_edge_torch/testing/__init__.py,sha256=_yGgvnBZWb7T3IN3mc4x1sS4vM96HZwM8pwIcPG
|
|
|
270
270
|
ai_edge_torch/testing/export.py,sha256=k5mGDGzwc23Z4zaIVDs8CNh-oOt64gsf9MS9NjhbPy4,3293
|
|
271
271
|
ai_edge_torch/testing/model_coverage/__init__.py,sha256=5P8J6Zk5YYtDvTBucFvB9NGSRI7Gw_24WnrbhXgycEE,765
|
|
272
272
|
ai_edge_torch/testing/model_coverage/model_coverage.py,sha256=UPB448aMDUyC0HNYVqio2rcJPnDN0tBQMP08J6vPYew,4718
|
|
273
|
-
ai_edge_torch_nightly-0.7.0.
|
|
274
|
-
ai_edge_torch_nightly-0.7.0.
|
|
275
|
-
ai_edge_torch_nightly-0.7.0.
|
|
276
|
-
ai_edge_torch_nightly-0.7.0.
|
|
277
|
-
ai_edge_torch_nightly-0.7.0.
|
|
273
|
+
ai_edge_torch_nightly-0.7.0.dev20251102.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
|
|
274
|
+
ai_edge_torch_nightly-0.7.0.dev20251102.dist-info/METADATA,sha256=11IrbboYIdBWLBmcqSzRhodc6pY82AQLqLfE-WcjW88,2399
|
|
275
|
+
ai_edge_torch_nightly-0.7.0.dev20251102.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
|
|
276
|
+
ai_edge_torch_nightly-0.7.0.dev20251102.dist-info/top_level.txt,sha256=5KXRaF2hwkApYxf7Y8y_tVb9aulGTlbOoNdbx1aKRkE,14
|
|
277
|
+
ai_edge_torch_nightly-0.7.0.dev20251102.dist-info/RECORD,,
|
|
File without changes
|