From e332f9181281bad44149b4ab9ee237153792dbab Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Thu, 15 May 2025 07:29:32 -0700 Subject: [PATCH] Sesame --- pyproject.toml | 4 ++-- unsloth/models/_utils.py | 2 +- unsloth/models/mapper.py | 4 ++++ 3 files changed, 7 insertions(+), 3 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index d438c83d67..4cadd3aa67 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -37,7 +37,7 @@ triton = [ ] huggingface = [ - "unsloth_zoo>=2025.5.5", + "unsloth_zoo>=2025.5.6", "packaging", "tyro", "transformers==4.51.3,!=4.47.0", @@ -381,7 +381,7 @@ colab-ampere-torch220 = [ "flash-attn>=2.6.3", ] colab-new = [ - "unsloth_zoo>=2025.5.5", + "unsloth_zoo>=2025.5.6", "packaging", "tyro", "transformers==4.51.3,!=4.47.0", diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py index 882de28cba..118f4f0535 100644 --- a/unsloth/models/_utils.py +++ b/unsloth/models/_utils.py @@ -12,7 +12,7 @@ # See the License for the specific language governing permissions and # limitations under the License. -__version__ = "2025.5.3" +__version__ = "2025.5.4" __all__ = [ "SUPPORTS_BFLOAT16", diff --git a/unsloth/models/mapper.py b/unsloth/models/mapper.py index d723fc4bd6..fd8b2e60d6 100644 --- a/unsloth/models/mapper.py +++ b/unsloth/models/mapper.py @@ -817,6 +817,10 @@ __INT_TO_FLOAT_MAPPER = \ "microsoft/Phi-4-mini-reasoning", "unsloth/phi-4-mini-reasoning-bnb-4bit", ), + "unsloth/csm-1b" : ( + "unsloth/csm-1b", + "sesame/csm-1b", + ), } INT_TO_FLOAT_MAPPER = {}