litert-torch-nightly 0.9.0.dev20260128__py3-none-any.whl → 0.9.0.dev20260129__py3-none-any.whl

litert_torch/cli.py

@@ -23,7 +23,7 @@ from litert_torch.generative.export_hf import export as hf_export_lib
 class CLI:
 
   def __init__(self):
-    self.hf_export = hf_export_lib.export
+    self.export_hf = hf_export_lib.export
 
 
 def main():

litert_torch/generative/export_hf/core/export_lib.py

@@ -48,12 +48,6 @@ def verify_model_compatibility(model, model_config, text_model_config):
     print(utils.ERROR_MESSAGE)
     print('Dynamic and longrope are not supported yet.')
     raise NotImplementedError('Dynamic and longrope are not supported yet.')
-  if getattr(text_model_config, 'rope_scaling', None) is not None:
-    print(utils.ERROR_MESSAGE)
-    print(
-        'rope_scaling is not supported yet, the model exported will have wrong'
-        ' results and is for BENCHMARKING ONLY.'
-    )
   can_compile_fullgraph = getattr(model, '_can_compile_fullgraph', None)
   if can_compile_fullgraph is None:
     print(utils.WARNING_MESSAGE)

litert_torch/generative/export_hf/export_main.py

@@ -34,4 +34,4 @@ def main(_):
 
 
 if __name__ == '__main__':
-  fire.run()
+  main()

litert_torch/version.py

@@ -15,4 +15,4 @@
 
 # The next version of litert-torch.
 # The minor version code should be bumped after every release.
-__version__ = "0.9.0.dev20260128"
+__version__ = "0.9.0.dev20260129"

{litert_torch_nightly-0.9.0.dev20260128.dist-info → litert_torch_nightly-0.9.0.dev20260129.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: litert-torch-nightly
-Version: 0.9.0.dev20260128
+Version: 0.9.0.dev20260129
 Summary: Support PyTorch model conversion with LiteRT.
 Home-page: https://github.com/google-ai-edge/litert-torch
 Keywords: On-Device ML,AI,Google,TFLite,LiteRT,PyTorch,LLMs,GenAI

{litert_torch_nightly-0.9.0.dev20260128.dist-info → litert_torch_nightly-0.9.0.dev20260129.dist-info}/RECORD

@@ -1,9 +1,9 @@
 litert_torch/__init__.py,sha256=jgEpTtwnhxMhPGPTRvHJR7pzx6WU_mLbA-G1LjO7fnE,1279
 litert_torch/_config.py,sha256=zDnki83sBsQzDAea6bvzwccylWHnPUzbEyGGRh6B14w,2526
-litert_torch/cli.py,sha256=TiguLo2O3_wY8cCnKnbXtUvGH4lzyjeSgsOnHsKd9Gg,984
+litert_torch/cli.py,sha256=TiAUbgbWm3ecTUJtJ1_hjKJuC1LrG-Qwnm8_zws-sVY,984
 litert_torch/conftest.py,sha256=gYmFrsR4c_fjIidbyrDnek26yS0crDP6-UoyMvy-WFg,757
 litert_torch/model.py,sha256=KXFTyyfPM6AnP0JoSwsTqQR3lUQbMkTGSr3dUsfQ5Jk,5635
-litert_torch/version.py,sha256=mHpimDjWXUBj0_q1ks2U5uwQ-hARKooCxFvQ3Hp4RPU,804
+litert_torch/version.py,sha256=gWoiaB9PVwbFzTApCzIlPdGP1qzvsc0kWvKlx9Gqf68,804
 litert_torch/_convert/__init__.py,sha256=qdLdbj5NjhNG-QgY5O_8TzOr2XaDoWvmdY9JNPStQmw,670
 litert_torch/_convert/conversion.py,sha256=NuQEphyYp3W19IKvyTWo9pe7zt1-XmWM4zU9PDkUm54,6108
 litert_torch/_convert/conversion_utils.py,sha256=MWpB-3eN-rvQzTtXsPL30cDIK431SQuwvw3ia2K2ONM,2158
@@ -180,14 +180,14 @@ litert_torch/generative/examples/tiny_llama/verify_util.py,sha256=FKMC6Olex6bJbB
 litert_torch/generative/export_hf/__init__.py,sha256=5xWIp2ziIwapcZcjSKfeaFgBnIooa8ckhTQ7mazZC3c,670
 litert_torch/generative/export_hf/__main__.py,sha256=8VuBDkZ2sL-q2XdQ45qwzeHQk39-MM_6TdkxOU_23xE,782
 litert_torch/generative/export_hf/export.py,sha256=FGjfwyWBNzGaBxTZHq-w0qq1W9BG0SuM1OV67G8nmfI,2686
-litert_torch/generative/export_hf/export_main.py,sha256=1dmw-V0UM2yt9XnIbpw87-YuTlr-qYjethMuyW9-yFA,1187
+litert_torch/generative/export_hf/export_main.py,sha256=bQidNXz0MEP_gil86LSfnpCW0pUiqZq2-F9ZOrSb3Yk,1183
 litert_torch/generative/export_hf/core/__init__.py,sha256=5xWIp2ziIwapcZcjSKfeaFgBnIooa8ckhTQ7mazZC3c,670
 litert_torch/generative/export_hf/core/attention.py,sha256=FFqq5TZ-V4Bdn2jb06_VWjAzVj1dlDuV1Llkftf_J54,2454
 litert_torch/generative/export_hf/core/attention_test.py,sha256=RevOczfPncmbIBths1bjYIWFbCHAZ3JK367NPt56dDs,4923
 litert_torch/generative/export_hf/core/cache.py,sha256=pfWh2SACdhNY2of2Z8KJC0wrSQ2jrkXgPHWe7PSEiuU,10263
 litert_torch/generative/export_hf/core/cache_base.py,sha256=FXMm9B8nDwC8uTyLmuBnYKLTnNtoeGN8gUnWwDCcH08,1714
 litert_torch/generative/export_hf/core/cache_test.py,sha256=y-v-oOGtRNPGWRfIfW3FcpDxvJbzrBU6Pb2o66FkUzU,6203
-litert_torch/generative/export_hf/core/export_lib.py,sha256=Nvg3QiYRZcMiQd7du7w5vohazjLlZJW6YFk_WAbIpAs,12249
+litert_torch/generative/export_hf/core/export_lib.py,sha256=4Nr_kaUgtukCMbcYv5MnJmdlrUCau1FeyVR7o-8Y3SU,12005
 litert_torch/generative/export_hf/core/exportable_module.py,sha256=XEqsV9M34OP-_vsxH7bnmxSCD6erAPl0a9I9JQM7v6k,8305
 litert_torch/generative/export_hf/core/litert_lm_builder.py,sha256=f8Q2ifVyt65V-kRL0X9FRpQNKIer0R_Yx2lECZTMGPU,7965
 litert_torch/generative/export_hf/core/patches.py,sha256=i1fzs0anIFbBH-Q_PwCtp9VKXy64olJKwnGpnJUjkEo,1815
@@ -318,9 +318,9 @@ litert_torch/testing/__init__.py,sha256=AfYP1HwTYSQmupveonEHCDV5dEyshzUgbwUrCUhb
 litert_torch/testing/export.py,sha256=3dR6oxnrdtX0MfqAfMv233cf3sHA4e0F2TBQotoo8xc,3292
 litert_torch/testing/model_coverage/__init__.py,sha256=uPXeAhWiD1O0aMDLCX7FTOSNQiea8yOtoIYPCuHEAG4,763
 litert_torch/testing/model_coverage/model_coverage.py,sha256=EPCI7PbNPb7GV28lo3qQvFdzJwJ_ZDrbCGdpeiBZhVo,4715
-litert_torch_nightly-0.9.0.dev20260128.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
-litert_torch_nightly-0.9.0.dev20260128.dist-info/METADATA,sha256=F-LavniKQ7N46UMAKuq-94y75WvGAlcbfwEcjrKVUTQ,2463
-litert_torch_nightly-0.9.0.dev20260128.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
-litert_torch_nightly-0.9.0.dev20260128.dist-info/entry_points.txt,sha256=roYAi9hp0uYrMudMR59hGNF2pz0TSAtqNl4vQLJzxnE,55
-litert_torch_nightly-0.9.0.dev20260128.dist-info/top_level.txt,sha256=mGrsl2SYcjQSLBJX4ZXrHnFqHZe6QLRR7uk0tLfzwfM,13
-litert_torch_nightly-0.9.0.dev20260128.dist-info/RECORD,,
+litert_torch_nightly-0.9.0.dev20260129.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
+litert_torch_nightly-0.9.0.dev20260129.dist-info/METADATA,sha256=WJXxJ3cwIDOIAmpPpXpnVr6PAszL8rRR6uHgMlJqDRs,2463
+litert_torch_nightly-0.9.0.dev20260129.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
+litert_torch_nightly-0.9.0.dev20260129.dist-info/entry_points.txt,sha256=roYAi9hp0uYrMudMR59hGNF2pz0TSAtqNl4vQLJzxnE,55
+litert_torch_nightly-0.9.0.dev20260129.dist-info/top_level.txt,sha256=mGrsl2SYcjQSLBJX4ZXrHnFqHZe6QLRR7uk0tLfzwfM,13
+litert_torch_nightly-0.9.0.dev20260129.dist-info/RECORD,,