August-2025-v2

unslothai/unslothAugust-2025-v2Aug 28, 2025by shimmyshimmer

AI Summary

Introduces Unsloth Flex Attention for gpt-oss training, enabling significantly longer contexts and lower VRAM usage compared to Flash Attention 3.

Key Highlights

  • Flex Attention support for gpt-oss training
  • >8x longer context, >50% less VRAM, >1.5x faster than FA3
  • Supports 60K context length on 80GB VRAM
  • Export models to llama.cpp, vLLM, or HF
  • Fixed gpt-oss training losses on float16 GPUs

New Features

  • Flex Attention
  • QAT + LoRA support
  • Torch 2.8 support
  • save_pretrained_torchao

Full Release Notes

We’re excited to introduce Unsloth Flex Attention support for OpenAI gpt-oss training that enables **>8× longer context** lengths, **>50% less VRAM** usage and **>1.5× faster** training compared to all implementations including those using Flash Attention 3 (FA3). Unsloth Flex Attention makes it possible to train with a 60K context length on just 80GB of VRAM for BF16 LoRA. Also:
- You can now export/save your QLoRA fine-tuned gpt-oss model to llama.cpp, vLLM, or HF.
- We fixed gpt-oss training losses going to infinity on float16 GPUs (like T4 Colab)
- We fixed gpt-oss implementation issues, most notably ensuring that `swiglu_limit = 7.0` is properly applied during MXFP4 inference [in transformers](https://github.com/huggingface/transformers/pull/40197)
- Unsloth Flex Attention scales with context, longer sequences yield bigger savings in both VRAM and training time

Full details in our blogpost: https://docs.unsloth.ai/basics/long-context-gpt-oss-training

<a href="https://docs.unsloth.ai/basics/long-context-gpt-oss-training"><img src="https://github.com/user-attachments/assets/ea47ed4a-7a29-4c75-b9ac-2de4e13c91e5" width="600"></a>

## What's Changed
* Add Qwen3 Instruct / Thinking chat templates by @Etherll in https://github.com/unslothai/unsloth/pull/3110
* Add Qwen3 4B to mapper.py by @Etherll in https://github.com/unslothai/unsloth/pull/3120
* Nightly by @danielhanchen in https://github.com/unslothai/unsloth/pull/3148
* Fix GPT OSS by @danielhanchen in https://github.com/unslothai/unsloth/pull/3154
* Nightly by @danielhanchen in https://github.com/unslothai/unsloth/pull/3169
* Update Blackwell install instructions for latest vLLM release by @qingy1337 in https://github.com/unslothai/unsloth/pull/3175
* Fix potential generator exhaustion bug in model loading file detection by @rolandtannous in https://github.com/unslothai/unsloth/pull/3167
* Fix vision model GGUF quantization_method error type by @rolandtannous in https://github.com/unslothai/unsloth/pull/3173
* Replace back ticks with single quotes by @rnowling in https://github.com/unslothai/unsloth/pull/3157
* Fix original_push_to_hub fallback by @Thiraput01 in https://github.com/unslothai/unsloth/pull/3115
* Add support for QAT + LoRA by @andrewor14 in https://github.com/unslothai/unsloth/pull/2976
* Bug fixes by @danielhanchen in https://github.com/unslothai/unsloth/pull/3180
* Torch 2.8 by @danielhanchen in https://github.com/unslothai/unsloth/pull/3186
* Fix extras transformers typo in pyproject.toml by @parth2510 in https://github.com/unslothai/unsloth/pull/3187
* Bug fixes by @danielhanchen in https://github.com/unslothai/unsloth/pull/3195
* allow torch.float32 dtype in FastLanguageModel by @mmathew23 in https://github.com/unslothai/unsloth/pull/3204
* fix is casual for qwen3 by @leizhenyuan in https://github.com/unslothai/unsloth/pull/3213
* Support `model.save_pretrained_torchao` by @jerryzh168 in https://github.com/unslothai/unsloth/pull/3111
* Fix gemma-3n by @mmathew23 in https://github.com/unslothai/unsloth/pull/3219
* Handle transformers move to dtype from torch_dtype by @mmathew23 in https://github.com/unslothai/unsloth/pull/3225
* chore: Fix Typos by @DefiWimar7 in https://github.com/unslothai/unsloth/pull/3224

## New Contributors
* @rnowling made their first contribution in https://github.com/unslothai/unsloth/pull/3157
* @Thiraput01 made their first contribution in https://github.com/unslothai/unsloth/pull/3115
* @andrewor14 made their first contribution in https://github.com/unslothai/unsloth/pull/2976
* @parth2510 made their first contribution in https://github.com/unslothai/unsloth/pull/3187
* @jerryzh168 made their first contribution in https://github.com/unslothai/unsloth/pull/3111
* @DefiWimar7 made their first contribution in https://github.com/unslothai/unsloth/pull/3224

**Full Changelog**: https://github.com/unslothai/unsloth/compare/August-2025...August-2025-v2