ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Devstral, MedGemma

Browse source
This commit is contained in:
Daniel Han 2025-05-21 07:35:36 -07:00
commit 911a1def95
3 changed files with 18 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
pyproject.toml
View file

@ -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",

2
unsloth/models/_utils.py
View file

@ -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",

15
unsloth/models/mapper.py
View file

@ -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 = {}