unsloth/unsloth/models
2026-05-15 11:03:34 -07:00
..
__init__.py Revert "feat: Add Mixtral model support" 2026-03-13 22:38:49 -07:00
_utils.py Restore Flash > SDPA > Flex priority for non-gemma3 models (#5455) 2026-05-15 12:43:49 -05:00
cohere.py Fix forward compatibility with transformers 5.x (#4752) 2026-04-01 06:04:03 -07:00
dpo.py Formatting & bug fixes (#3563) 2025-11-07 06:00:22 -08:00
falcon_h1.py Fix forward compatibility with transformers 5.x (#4752) 2026-04-01 06:04:03 -07:00
gemma.py Fix/pr 3699 leftpad prefill main (#4100) 2026-02-25 07:21:04 -08:00
gemma2.py Fix forward compatibility with transformers 5.x (#4752) 2026-04-01 06:04:03 -07:00
gemma4_moe_4bit.py [pre-commit.ci] auto fixes from pre-commit.com hooks 2026-05-15 03:50:06 +00:00
glm4_moe.py [MoE] Improve moe kernels for unsloth fine tuning (#3812) 2026-02-05 06:03:25 -08:00
granite.py Fix forward compatibility with transformers 5.x (#4752) 2026-04-01 06:04:03 -07:00
llama.py guardrail: detect silent 4-bit / 8-bit quantization bypass (#5344) 2026-05-15 03:45:41 +00:00
llama4.py Qwen 3 2025-05-02 03:09:44 -07:00
loader.py Restrict flash attn to <=256 head dim. Consolidate attn impl checks (#5051) 2026-04-16 09:00:17 -05:00
loader_utils.py studio: reuse HF cached repo casing to prevent duplicate downloads (#4822) 2026-04-03 05:48:24 -07:00
mapper.py Re-apply #4939: updated models template mappers (#4950) 2026-04-15 07:52:12 -07:00
mistral.py Fix Mistral DPO/preference training crash on non-xformers platforms (e.g. Intel XPU) (#4889) 2026-04-09 04:38:44 -07:00
qwen2.py Revert "[pre-commit.ci] auto fixes from pre-commit.com hooks" 2025-12-01 07:24:58 -08:00
qwen3.py Fix forward compatibility with transformers 5.x (#4752) 2026-04-01 06:04:03 -07:00
qwen3_moe.py Fix correctness bugs across multiple model files (#3813) 2026-01-01 02:36:33 -08:00
rl.py add UNSLOTH_ALLOW_CPU=1 path for CPU-only CI (#5429) 2026-05-14 20:27:14 -07:00
rl_replacements.py security: NOT affected by Mini Shai-Hulud (May-12 wave) -- forward-looking hardening only (#5397) 2026-05-13 04:58:12 -07:00
sentence_transformer.py Fix: Add missing utf-8 encoding to text-mode file operations (#5356) 2026-05-14 18:15:27 +04:00
vision.py Merge branch 'fix-issue-5344-quantization-guardrail' into feat-gemma4-moe-4bit-swap 2026-05-15 00:12:07 -07:00