unsloth/tests
Daniel Han 1e20366a26 ci(mlx): fresh-process reloads + soft-skip GGUF on llama.cpp limitation
Re-apply the subcommand restructure that was lost during the earlier
rebase conflict (the linter pre-commit on the remote re-formatted the
single-function version, so my checkout --ours kept the wrong copy).
Adds:

  * argparse subcommands `train` and `reload --format X --dir D` so
    each reload runs in a FRESH Python process the way real users
    hit the cold-start path.
  * Per-phase Phase() context manager records elapsed wall-clock,
    peak GPU memory (mx.metal.get_peak_memory), and peak RSS
    (resource.getrusage) into a metrics dict written to
    {train,lora_reload,merged_reload,gguf_reload}_metrics.json
    next to the saved dir for cross-CI regression detection.
  * batch_size=2, gradient_accumulation_steps=3 (was 2/1) so the
    7-step run sees 42 sequences total.
  * GGUF save is best-effort. unsloth-zoo#627 fixed the
    NotImplementedError on Apple Silicon, but llama.cpp's
    convert_hf_to_gguf currently asserts on the gemma-3-270m
    tokenizer vocab (`max(vocab IDs) >= vocab_size`). That's a
    downstream llama.cpp limitation, not an unsloth_zoo bug, so the
    train step records gguf_supported=false + the reason instead of
    raising, and the GGUF reload step emits a workflow warning and
    exits 0. The LoRA + merged_16bit reload assertions remain the
    gating signal.

The earlier-draft LoRA workaround that copied base config.json into
the LoRA save dir is removed; unsloth-zoo#627 makes
FastMLXModel.from_pretrained(lora_dir) work on the saved adapter
directory directly (the failing run before #627 confirmed the bug,
the run after #627 lands shows the adapter is detected and the base
model is pulled from adapter_config.json:base_model_name_or_path).
2026-05-07 04:12:21 +00:00
..
python Add Studio PR-time CI: pin enforcement, frontend, backend, wheel smoke (#5298) 2026-05-06 04:41:57 -07:00
qlora Revert "[pre-commit.ci] auto fixes from pre-commit.com hooks" 2025-12-01 07:24:58 -08:00
saving chore: fix typo cleanup across tests and backend strings (#5152) 2026-04-24 12:51:27 +01:00
sh Default Studio host to 127.0.0.1 and prompt before auto-start (#5267) 2026-05-04 13:03:16 +04:00
studio ci(mlx): fresh-process reloads + soft-skip GGUF on llama.cpp limitation 2026-05-07 04:12:21 +00:00
utils feat: Add cactus QAT scheme support (#4679) 2026-04-15 07:40:03 -07:00
__init__.py Qwen 3, Bug Fixes (#2445) 2025-04-30 22:38:39 -07:00
conftest.py Add Studio PR-time CI: pin enforcement, frontend, backend, wheel smoke (#5298) 2026-05-06 04:41:57 -07:00
run_all.sh fix: add tokenizers to no-torch deps and TORCH_CONSTRAINT for arm64 macOS py313+ (#4748) 2026-04-01 06:12:17 -07:00
test_cli_export_unpacking.py studio: stream export worker output into the export dialog (#4897) 2026-04-14 08:55:43 -07:00
test_gemma4_chat_template.py update gema4 chat templates (#5116) 2026-04-22 09:04:08 -07:00
test_get_model_name.py feat: Add support for OLMo-3 model (#4678) 2026-04-15 07:39:11 -07:00
test_loader_glob_skip.py Add unit tests for HfFileSystem glob skip guard (#4854) 2026-04-06 08:54:36 -07:00
test_model_registry.py Revert "[FIX] Vllm guided decoding params (#3662)" 2025-12-01 05:43:45 -08:00
test_peft_weight_converter_compat.py Patch checkpoint reload init functions to strip unsupported args (#5167) 2026-04-29 02:50:49 -07:00
test_raw_text.py Add Studio PR-time CI: pin enforcement, frontend, backend, wheel smoke (#5298) 2026-05-06 04:41:57 -07:00
test_resolve_model_class.py fix: guard resolve_model_class fallback against unresolvable transformers AutoModel entries (#5155) 2026-04-24 05:59:17 -07:00
test_studio_install_workspace_guard.py install: support STUDIO_HOME / UNSLOTH_STUDIO_HOME for custom install paths (#5190) 2026-05-05 23:17:40 -07:00
test_studio_root_resilience.py install: support STUDIO_HOME / UNSLOTH_STUDIO_HOME for custom install paths (#5190) 2026-05-05 23:17:40 -07:00