lalamo 0.5.1__py3-none-any.whl → 0.5.2__py3-none-any.whl

lalamo/__init__.py

@@ -10,7 +10,7 @@ from lalamo.message_processor import (
 )
 from lalamo.model_import import ModelSpec, import_model
 
-__version__ = "0.5.1"
+__version__ = "0.5.2"
 
 __all__ = [
     "AssistantMessage",

lalamo/model_import/decoder_configs/huggingface/gemma3.py

@@ -1,5 +1,5 @@
 from collections.abc import Mapping
-from dataclasses import dataclass
+from dataclasses import dataclass, field
 from typing import Literal
 
 import jax.numpy as jnp
@@ -159,6 +159,7 @@ class HFGemma3TextConfigRaw:
 @dataclass(frozen=True)
 class HFGemma3TextConfig(HFGemma3TextConfigRaw, HuggingFaceConfig):
     torch_dtype: Literal["bfloat16", "float16", "float32"] = "bfloat16"
+    eos_token_id: int | list[int] = field(default_factory=list)
 
 
 @dataclass(frozen=True)

{lalamo-0.5.1.dist-info → lalamo-0.5.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lalamo
-Version: 0.5.1
+Version: 0.5.2
 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.5.1.dist-info → lalamo-0.5.2.dist-info}/RECORD

@@ -1,4 +1,4 @@
-lalamo/__init__.py,sha256=AtCHHyq3yCP2A-NnctlQc-gSTUFOfDqP2PpiJX0ay28,487
+lalamo/__init__.py,sha256=3P0i5yWyx-A6xFJNXobXqcLYXbgpLKPr4tL-x3OAJto,487
 lalamo/common.py,sha256=5NUFD26yQgOnEEk3LaQnce8n-VwJxILkEpFesHZhtQU,3820
 lalamo/language_model.py,sha256=ZnNwqxT8yCfIE0TSaF8Vhva8GRHzDzEoUb95VjAfOtM,14510
 lalamo/main.py,sha256=YNO7s0H04Jg5OYmKEGmdIJLzW1W_-uJj0TxJY_Tsc0M,19028
@@ -21,7 +21,7 @@ lalamo/model_import/decoder_configs/executorch.py,sha256=tyJ-oP8wezOHBGL7Y9Iy3np
 lalamo/model_import/decoder_configs/huggingface/__init__.py,sha256=m17Pz3pw7lHXrcw3FGE15TNUu-Qk7m9CYQSXfS3CWD0,575
 lalamo/model_import/decoder_configs/huggingface/common.py,sha256=BJekEs9iStsygVa9QcPItok5AQIq8PvMZf7xAmykl-w,3043
 lalamo/model_import/decoder_configs/huggingface/gemma2.py,sha256=0pO4IinzqB7hLoEUMlVnJfN3ncerTgZKIrMZPs3soR8,4562
-lalamo/model_import/decoder_configs/huggingface/gemma3.py,sha256=e1Vln08e5V74jVuX5ddzDsEU4fnwt2XT-rv0ho1_kWM,7146
+lalamo/model_import/decoder_configs/huggingface/gemma3.py,sha256=r8Nfy-9EPXhIkhJkvx0P-7NFLMaku_LJIEhx3HziShk,7217
 lalamo/model_import/decoder_configs/huggingface/gpt_oss.py,sha256=8TWptvJzVJtEtyWrkKaJA6LLORSKLolxv3gwLtDC0uA,7248
 lalamo/model_import/decoder_configs/huggingface/llama.py,sha256=amMhyN_QEscFmOvjmghcBg9VXBDqE-mpp0Y5Gj0tSp0,6735
 lalamo/model_import/decoder_configs/huggingface/llamba.py,sha256=dJVD_84suhF5sQLRvlsfMDmrrTNhIR6AdMGOm54rC6U,5482
@@ -72,9 +72,9 @@ lalamo/speculator/common.py,sha256=PudF_gkpe5_nQ-57sAC-foE1xCy_H2Axh5KwRoA86lo,5
 lalamo/speculator/inference.py,sha256=4E7_8jwLwg1bvUUH59cLxjD73-IJ9xv58yCwYMFa1r4,3101
 lalamo/speculator/ngram.py,sha256=3O9akE2IGN_7Mp6l_zwhUpdNvjZB0rAlQm51EuX_-rY,5869
 lalamo/speculator/utils.py,sha256=K5qDGEdWYD47C1xLJQ3bamIHAG3-VEuANf-821VeanE,1688
-lalamo-0.5.1.dist-info/licenses/LICENSE,sha256=diHRfjSEJHD1nnEeMIfMRCjR3UERf8bT3eseD6b1ayA,1072
-lalamo-0.5.1.dist-info/METADATA,sha256=xHWGxP4GYjQkOaQpOxKRJNINprD2YiJXZb2tDVvDMRA,3146
-lalamo-0.5.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-lalamo-0.5.1.dist-info/entry_points.txt,sha256=qli7qTfnBk5WP10rOGXXEckHMtt-atJMDWd8jN89Uks,43
-lalamo-0.5.1.dist-info/top_level.txt,sha256=VHvWL5JN5XRG36NsN_MieJ7EwRihEOrEjyDaTdFJ-aI,7
-lalamo-0.5.1.dist-info/RECORD,,
+lalamo-0.5.2.dist-info/licenses/LICENSE,sha256=diHRfjSEJHD1nnEeMIfMRCjR3UERf8bT3eseD6b1ayA,1072
+lalamo-0.5.2.dist-info/METADATA,sha256=bEhxBxyO5vmbF5KyrEMRSATIzJ55M9HfunCkllcC-VY,3146
+lalamo-0.5.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+lalamo-0.5.2.dist-info/entry_points.txt,sha256=qli7qTfnBk5WP10rOGXXEckHMtt-atJMDWd8jN89Uks,43
+lalamo-0.5.2.dist-info/top_level.txt,sha256=VHvWL5JN5XRG36NsN_MieJ7EwRihEOrEjyDaTdFJ-aI,7
+lalamo-0.5.2.dist-info/RECORD,,