From 911a1def95cdbdfafee230d8a2e759a7714f914d Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Wed, 21 May 2025 07:35:36 -0700 Subject: [PATCH] Devstral, MedGemma --- pyproject.toml | 4 ++-- unsloth/models/_utils.py | 2 +- unsloth/models/mapper.py | 15 +++++++++++++++ 3 files changed, 18 insertions(+), 3 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 558abd9f1a..7fb572b735 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -37,7 +37,7 @@ triton = [ ] huggingface = [ - "unsloth_zoo>=2025.5.7", + "unsloth_zoo>=2025.5.8", "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.7", + "unsloth_zoo>=2025.5.8", "packaging", "tyro", "transformers==4.51.3,!=4.47.0", diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py index 747858d011..964e874c58 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.6" +__version__ = "2025.5.7" __all__ = [ "SUPPORTS_BFLOAT16", diff --git a/unsloth/models/mapper.py b/unsloth/models/mapper.py index e50a5a877d..50f0f7d7fb 100644 --- a/unsloth/models/mapper.py +++ b/unsloth/models/mapper.py @@ -849,6 +849,21 @@ __INT_TO_FLOAT_MAPPER = \ "unsloth/Llama-OuteTTS-1.0-1B", "OuteAI/Llama-OuteTTS-1.0-1B", ), + "unsloth/medgemma-4b-it-unsloth-bnb-4bit" : ( + "unsloth/medgemma-4b-it", + "google/medgemma-4b-it", + "unsloth/medgemma-4b-it-bnb-4bit", + ), + "unsloth/medgemma-27b-text-it-unsloth-bnb-4bit" : ( + "unsloth/medgemma-27b-text-it", + "google/medgemma-27b-text-it", + "unsloth/medgemma-27b-text-it-bnb-4bit", + ), + "unsloth/Devstral-Small-2505-unsloth-bnb-4bit" : ( + "unsloth/Devstral-Small-2505", + "mistralai/Devstral-Small-2505", + "unsloth/Devstral-Small-2505-bnb-4bit", + ), } INT_TO_FLOAT_MAPPER = {}