lalamo 0.3.0__py3-none-any.whl → 0.3.1__py3-none-any.whl

lalamo/__init__.py

@@ -1,7 +1,7 @@
 from lalamo.model_import import REPO_TO_MODEL, ModelSpec, import_model
 from lalamo.modules import Decoder
 
-__version__ = "0.3.0"
+__version__ = "0.3.1"
 
 __all__ = [
     "REPO_TO_MODEL",

lalamo/model_import/model_specs/common.py

@@ -10,7 +10,6 @@ from jaxtyping import Array, DTypeLike
 from safetensors.flax import load_file as load_safetensors
 
 from lalamo.model_import.decoder_configs import ForeignConfig
-from lalamo.modules.torch_interop import torch_to_jax
 from lalamo.quantization import QuantizationMode
 from lalamo.utils import MapDictValues
 
@@ -41,6 +40,8 @@ class WeightsType(Enum):
 
         import torch
 
+        from lalamo.modules.torch_interop import torch_to_jax
+
         torch_weights = torch.load(filename, map_location="cpu", weights_only=True)
         return MapDictValues(lambda v: cast_if_float(torch_to_jax(v), float_dtype), torch_weights)
 

{lalamo-0.3.0.dist-info → lalamo-0.3.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lalamo
-Version: 0.3.0
+Version: 0.3.1
 Summary: JAX library for optimization and export of models for use with the UZU inference engine.
 Requires-Python: <4,>=3.12
 Description-Content-Type: text/markdown

{lalamo-0.3.0.dist-info → lalamo-0.3.1.dist-info}/RECORD

@@ -1,4 +1,4 @@
-lalamo/__init__.py,sha256=5ZLxT3qwA2jPRZ8jgjR0c3mgeDPe_VlgQXOIxYE6FgY,217
+lalamo/__init__.py,sha256=BBoNVaSUlCj5z4W_8Mo-GYkDZHVno1CmcZ370nWxDHs,217
 lalamo/common.py,sha256=5NUFD26yQgOnEEk3LaQnce8n-VwJxILkEpFesHZhtQU,3820
 lalamo/language_model.py,sha256=wZGucNCPP9ReHn6fYO5TlFHTXun1QIMwaxniq9z7K5Q,10639
 lalamo/main.py,sha256=Ta3ZW-Xw3uNPEVvb2YlBOH9UhhLDV7PZjRCg6giu0ao,12275
@@ -26,7 +26,7 @@ lalamo/model_import/loaders/common.py,sha256=kkugV-bMQlN1zvGHoj3uc7z0FbXKoMtXEBT
 lalamo/model_import/loaders/executorch.py,sha256=nSvpylK8QL3nBk78P3FabLoyA87E3kv5CCpMfvuZe6Q,8886
 lalamo/model_import/loaders/huggingface.py,sha256=C0spTy9-DhQ8U7bDAyGA_i9dKgUoLR2Dv58t20KUvZs,10590
 lalamo/model_import/model_specs/__init__.py,sha256=UsKmBzNzJHVQquZxWdjGgiznnORVklE-z9Nr9Ertfqc,964
-lalamo/model_import/model_specs/common.py,sha256=69_SPs0sIDkxDJ84xXZ05tRsz7WgXb4W5KPxOutPbT8,3773
+lalamo/model_import/model_specs/common.py,sha256=MwyQOe1QNxX5Cw-p1Emsrx6vckG3X4Xi56RACt2oOrg,3782
 lalamo/model_import/model_specs/deepseek.py,sha256=Umef93_ZBuq93yYsejIRNwj3udoln1gHfrv3SK5jyMo,417
 lalamo/model_import/model_specs/gemma.py,sha256=YGWM-J7jBEL16c_LZ9F_6dgeZnf4sEPLDKngl_3FbrE,1289
 lalamo/model_import/model_specs/huggingface.py,sha256=TEkU8y95_hmUWyF-Q5hn0dE2SvXbApghAsQwhWRu4D0,431
@@ -50,9 +50,9 @@ lalamo/modules/normalization.py,sha256=YecmA0hnVVYd473FzVgxunHYA377NWMKZRfuQNB8D
 lalamo/modules/rope.py,sha256=IlL4Jz4fw6eTKXXMls1BXBAgbuB3GnP0aVB9GrF1nA8,10012
 lalamo/modules/torch_interop.py,sha256=-mujd1zI4ec2w92Hd50RtDa0K3jl6ZSnPxc5r3Fp9nU,916
 lalamo/modules/utils.py,sha256=5QTdi34kEI5jix7TfTdB0mOYZbzZUul_T1y8eWCA6lQ,262
-lalamo-0.3.0.dist-info/licenses/LICENSE,sha256=diHRfjSEJHD1nnEeMIfMRCjR3UERf8bT3eseD6b1ayA,1072
-lalamo-0.3.0.dist-info/METADATA,sha256=_7HPR4KvK7S-ig9CpKc1K2rSLg3f9hHc8_XdvHkDB10,2855
-lalamo-0.3.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-lalamo-0.3.0.dist-info/entry_points.txt,sha256=qli7qTfnBk5WP10rOGXXEckHMtt-atJMDWd8jN89Uks,43
-lalamo-0.3.0.dist-info/top_level.txt,sha256=VHvWL5JN5XRG36NsN_MieJ7EwRihEOrEjyDaTdFJ-aI,7
-lalamo-0.3.0.dist-info/RECORD,,
+lalamo-0.3.1.dist-info/licenses/LICENSE,sha256=diHRfjSEJHD1nnEeMIfMRCjR3UERf8bT3eseD6b1ayA,1072
+lalamo-0.3.1.dist-info/METADATA,sha256=pkkxgoNYDIMLQuHsk4cqdaHDwBel5EQ9jWYMxF1jy40,2855
+lalamo-0.3.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+lalamo-0.3.1.dist-info/entry_points.txt,sha256=qli7qTfnBk5WP10rOGXXEckHMtt-atJMDWd8jN89Uks,43
+lalamo-0.3.1.dist-info/top_level.txt,sha256=VHvWL5JN5XRG36NsN_MieJ7EwRihEOrEjyDaTdFJ-aI,7
+lalamo-0.3.1.dist-info/RECORD,,