unsloth/unsloth
Datta Nimmaturi eb65ff9b2f vLLM FP8 quantized support for SFT/GRPO (#3414)
* Prefer loading model from pretrained instead of config

* Fixup FP8 forward pass and inference

* [WIP] Fix lora forwards

* Infer block size from weight shapes

* reconstruct weights from fp8 quants for lora matmul

* Return weight transpose and fix dtype

* Refactor FP8 operations

* Fix naming :)

* Saner compile

* do not depend on transformers

* [WIP] fix training

* Update comment

* fixup training

* use dequant kernel from deepseek

* Differentiate between fp8 and fbgemmfp8

* fixup differentiation b/w fp8 and fbgemm_fp8

* make inputs contiguous if required

* Improve dequant

* More robust handling

* Fixup backward pass for fbgemm_fp8

* refactor and use bf16 for dequant

* Use torch fp8 block matmul

* Disable torch block matmul for now

* safer import and cosmetics

* more cosmectics

* add torchao operations

* Spaceeeeeee
2025-10-16 03:07:05 -07:00
..
dataprep Synthetic Data updates (#3333) 2025-09-17 21:43:00 -07:00
kernels vLLM FP8 quantized support for SFT/GRPO (#3414) 2025-10-16 03:07:05 -07:00
models vLLM FP8 quantized support for SFT/GRPO (#3414) 2025-10-16 03:07:05 -07:00
registry Qwen 3, Bug Fixes (#2445) 2025-04-30 22:38:39 -07:00
utils Qwen 3, Bug Fixes (#2445) 2025-04-30 22:38:39 -07:00
__init__.py Versioning 2025-10-14 07:27:38 -07:00
_auto_install.py Update _auto_install.py 2025-08-19 03:20:37 -07:00
chat_templates.py [Part2] Reinstate llama.cpp Compatibility and GGUF Conversion with Multiple Quantizations and Automated Ollama Modelfile Creation (#3356) 2025-10-14 01:23:14 -07:00
import_fixes.py Update import_fixes.py 2025-10-14 05:40:21 -07:00
ollama_template_mappers.py [Part2] Reinstate llama.cpp Compatibility and GGUF Conversion with Multiple Quantizations and Automated Ollama Modelfile Creation (#3356) 2025-10-14 01:23:14 -07:00
save.py [Part2] Reinstate llama.cpp Compatibility and GGUF Conversion with Multiple Quantizations and Automated Ollama Modelfile Creation (#3356) 2025-10-14 01:23:14 -07:00
tokenizer_utils.py Update tokenizer_utils.py 2025-08-14 19:30:33 -07:00
trainer.py Initialize parent class in UnslothTrainingArguments constructor 2025-07-03 15:47:58 +00:00