unsloth/unsloth
Jerry Zhang 5c22f06a9a
Support saving locally in model.save_pretrained_torchao (#3263)
Summary:
Previously the test was not ran correctly and the save to local path is not tested
this PR added support for that and tries to test properly

Note: `python tests/saving/test_unsloth_save.py` doesn't run test

Test Plan:
pytest tests/saving/test_unsloth_save.py -k test_save_torchao

Reviewers:

Subscribers:

Tasks:

Tags:
2025-09-03 17:51:33 -07:00
..
dataprep Encoding UTF-8 2025-08-14 15:24:25 -07:00
kernels [Intel] make intel device support ROPE (#3164) 2025-09-03 04:39:57 -07:00
models chore: Fix Typos (#3246) 2025-09-03 04:38:41 -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 Bug fixes (#3195) 2025-08-20 07:39:43 -07:00
_auto_install.py Update _auto_install.py 2025-08-19 03:20:37 -07:00
chat_templates.py chore: Fix Typos (#3246) 2025-09-03 04:38:41 -07:00
import_fixes.py Bug fixes (#3195) 2025-08-20 07:39:43 -07:00
save.py Support saving locally in model.save_pretrained_torchao (#3263) 2025-09-03 17:51:33 -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