ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions
unsloth/unsloth/models
History
Datta Nimmaturi cff1e554fc [trl] Trl v0.28 (and above) rl fixes (#4156) 
* Refactor loss computation to include completion_mask

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Fixes for trl 0.28 and above

Remove sync/reload weights calls , remove vllm.LLM instantiation

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Refactor loss computation to include completion_mask

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Fixes for trl 0.28 and above

Remove sync/reload weights calls , remove vllm.LLM instantiation

* patch rpc in openenv for newer trl

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

---------

Co-authored-by: pluesclues <136766175+pluesclues@users.noreply.github.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2026-03-09 12:01:49 -04:00
..
__init__.py add FastSentenceTransformer for easily finetuning SentenceTransformer models (#3719) 2026-01-22 07:35:55 -08:00
_utils.py Bug fixed version 2026-03-08 06:39:19 -07:00
cohere.py Fix/pr 3699 leftpad prefill main (#4100) 2026-02-25 07:21:04 -08:00
dpo.py Formatting & bug fixes (#3563) 2025-11-07 06:00:22 -08:00
falcon_h1.py Fix/pr 3699 leftpad prefill main (#4100) 2026-02-25 07:21:04 -08:00
gemma.py Fix/pr 3699 leftpad prefill main (#4100) 2026-02-25 07:21:04 -08:00
gemma2.py Fix/pr 3699 leftpad prefill main (#4100) 2026-02-25 07:21:04 -08:00
glm4_moe.py [MoE] Improve moe kernels for unsloth fine tuning (#3812) 2026-02-05 06:03:25 -08:00
granite.py Fix/pr 3699 leftpad prefill main (#4100) 2026-02-25 07:21:04 -08:00
llama.py Fixup mapper issues and resolve properly (#4124) 2026-03-03 06:57:25 -08:00
llama4.py Qwen 3 2025-05-02 03:09:44 -07:00
loader.py Conditionally enable 4bit on CDNA for bitsandbytes>=v0.49.2 (#4161) 2026-03-07 01:33:40 -08:00
loader_utils.py Fixup mapper issues and resolve properly (#4124) 2026-03-03 06:57:25 -08:00
mapper.py Fixup mapper issues and resolve properly (#4124) 2026-03-03 06:57:25 -08:00
mistral.py Fix/pr 3699 leftpad prefill main (#4100) 2026-02-25 07:21:04 -08: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/pr 3699 leftpad prefill main (#4100) 2026-02-25 07:21:04 -08:00
qwen3_moe.py Fix correctness bugs across multiple model files (#3813) 2026-01-01 02:36:33 -08:00
rl.py [trl] Trl v0.28 (and above) rl fixes (#4156) 2026-03-09 12:01:49 -04:00
rl_replacements.py [trl] Trl v0.28 (and above) rl fixes (#4156) 2026-03-09 12:01:49 -04:00
sentence_transformer.py Fix EmbeddingGemma float16 NaN via FORCE_FLOAT32 for gemma3_text (#4014) 2026-02-10 01:40:13 -08:00
vision.py Fixup mapper issues and resolve properly (#4124) 2026-03-03 06:57:25 -08:00