unsloth/unsloth/models
oobabooga 2b06616a7e
Fix TrainingArguments silently disabling unsloth gradient checkpointing (#6829)
* Fix TrainingArguments silently disabling unsloth gradient checkpointing

* Cover loaded adapters and preserve explicit None in GC restore

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

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

---------

Co-authored-by: Lee Jackson <130007945+Imagineer99@users.noreply.github.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2026-07-03 16:35:02 +01:00
..
__init__.py Reduce and tighten code comments and docstrings repo-wide (#6095) 2026-06-08 23:09:51 -07:00
_utils.py fast_generate: clear error for vLLM-style inputs when fast_inference=False (#6786) 2026-07-03 08:16:32 -07:00
cohere.py Formatting: ruff line-length 100, kwarg-spacing passes, drop blank after short local imports (#6079) 2026-06-08 04:24:13 -07:00
diffusion.py Add FastDiffusionModel slow path for text-diffusion models (DiffusionGemma) (#6158) 2026-06-11 04:02:56 -07:00
dpo.py Formatting & bug fixes (#3563) 2025-11-07 06:00:22 -08:00
falcon_h1.py Remove unused FalconH1RMSNormGated import (#6728) 2026-06-28 02:43:34 -07:00
gemma.py Formatting: ruff line-length 100, kwarg-spacing passes, drop blank after short local imports (#6079) 2026-06-08 04:24:13 -07:00
gemma2.py Reduce and tighten code comments and docstrings repo-wide (#6095) 2026-06-08 23:09:51 -07:00
glm4_moe.py Reduce and tighten code comments and docstrings repo-wide (#6095) 2026-06-08 23:09:51 -07:00
granite.py Reduce and tighten code comments and docstrings repo-wide (#6095) 2026-06-08 23:09:51 -07:00
llama.py Fix TrainingArguments silently disabling unsloth gradient checkpointing (#6829) 2026-07-03 16:35:02 +01:00
llama4.py Qwen 3 2025-05-02 03:09:44 -07:00
loader.py Fix DDP crash from CPU-resident rotary inv_freq buffer (#6662) 2026-06-26 09:44:41 +01:00
loader_utils.py Fix DDP crash from CPU-resident rotary inv_freq buffer (#6662) 2026-06-26 09:44:41 +01:00
mapper.py Re-apply #4939: updated models template mappers (#4950) 2026-04-15 07:52:12 -07:00
mistral.py fix: use EMPTY_LOGITS on the fused-CE not-return_dict path (#2068) (#6482) 2026-06-23 01:26:55 -07:00
qwen2.py Formatting: ruff line-length 100, kwarg-spacing passes, drop blank after short local imports (#6079) 2026-06-08 04:24:13 -07:00
qwen3.py Reduce and tighten code comments and docstrings repo-wide (#6095) 2026-06-08 23:09:51 -07:00
qwen3_moe.py Reduce and tighten code comments and docstrings repo-wide (#6095) 2026-06-08 23:09:51 -07:00
rl.py Fix TrainingArguments silently disabling unsloth gradient checkpointing (#6829) 2026-07-03 16:35:02 +01:00
rl_replacements.py Fix TrainingArguments silently disabling unsloth gradient checkpointing (#6829) 2026-07-03 16:35:02 +01:00
sentence_transformer.py Reduce and tighten code comments and docstrings repo-wide (#6095) 2026-06-08 23:09:51 -07:00
vision.py Fix TrainingArguments silently disabling unsloth gradient checkpointing (#6829) 2026-07-03 16:35:02 +01:00