unsloth/tests/studio
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
..
install Add Studio PR-time CI: pin enforcement, frontend, backend, wheel smoke (#5298) 2026-05-06 04:41:57 -07:00
playwright_chat_ui.py [pre-commit.ci] auto fixes from pre-commit.com hooks 2026-05-07 04:09:29 +00:00
playwright_extra_ui.py [pre-commit.ci] auto fixes from pre-commit.com hooks 2026-05-07 04:09:29 +00:00
run_real_mlx_smoke.py ci(mlx): fresh-process reloads + soft-skip GGUF on llama.cpp limitation 2026-05-07 04:12:21 +00:00
test_cancel_atomicity.py Studio: make stop button actually stop generation (#5069) 2026-04-24 10:09:25 -07:00
test_cancel_id_wiring.py Studio: make stop button actually stop generation (#5069) 2026-04-24 10:09:25 -07:00
test_chat_preset_builtin_invariants.py [pre-commit.ci] auto fixes from pre-commit.com hooks 2026-05-06 11:55:39 +00:00
test_cli_repo_variant.py Studio: forward llama-server args from unsloth studio run , activate unsloth run , and allow passing model:quant to load models (#5271) 2026-05-04 17:08:04 +04:00
test_cli_run_alias.py Studio: forward llama-server args from unsloth studio run , activate unsloth run , and allow passing model:quant to load models (#5271) 2026-05-04 17:08:04 +04:00
test_cli_studio_defaults.py Default Studio host to 127.0.0.1 and prompt before auto-start (#5267) 2026-05-04 13:03:16 +04:00
test_export_output_path_contract.py feat(studio): MLX training tab on Apple Silicon (LoRA / full FT, VLM, export) (#5265) 2026-05-05 23:54:58 -07:00
test_hardware_dispatch_matrix.py [pre-commit.ci] auto fixes from pre-commit.com hooks 2026-05-07 03:33:12 +00:00
test_is_mlx_dispatch_gate.py feat(studio): MLX training tab on Apple Silicon (LoRA / full FT, VLM, export) (#5265) 2026-05-05 23:54:58 -07:00
test_llama_cpp_wall_clock_cap.py Studio: make stop button actually stop generation (#5069) 2026-04-24 10:09:25 -07:00
test_mlx_training_worker_behaviors.py feat(studio): MLX training tab on Apple Silicon (LoRA / full FT, VLM, export) (#5265) 2026-05-05 23:54:58 -07:00
test_stream_cancel_registration_timing.py Studio: make stop button actually stop generation (#5069) 2026-04-24 10:09:25 -07:00
test_studio_api_smoke.py [pre-commit.ci] auto fixes from pre-commit.com hooks 2026-05-07 04:09:29 +00:00
test_studio_gguf_export_script_pin.py Pin Studio GGUF export to llama.cpp's local convert script (#5275) 2026-05-05 04:03:28 -07:00
test_studio_text_descender_clipping.py Studio: Fix clipped model selector text descenders (#5210) 2026-04-29 02:51:25 -07:00