| from dota2tuned.model_profiles import apply_profile_env_overrides, resolve_model_profile | |
| def test_model_profiles_include_quality_and_sponsor_paths(): | |
| quality = resolve_model_profile("qwen3_30b_a3b_2507") | |
| sponsor = resolve_model_profile("minicpm4_1_8b") | |
| assert quality.base_model_id == "Qwen/Qwen3-30B-A3B-Instruct-2507" | |
| assert quality.sft_max_length == 4096 | |
| assert quality.modal_train_gpu == "H200" | |
| assert quality.lora_dropout == 0.0 | |
| assert "q_proj" in quality.lora_target_modules | |
| assert sponsor.base_model_id == "openbmb/MiniCPM4.1-8B" | |
| assert sponsor.hf_model_repo_id.endswith("minicpm4-1-8b-lora") | |
| def test_model_profile_env_overrides(monkeypatch): | |
| monkeypatch.setenv("BASE_MODEL_ID", "example/base") | |
| monkeypatch.setenv("HF_MODEL_REPO_ID", "example/adapter") | |
| monkeypatch.setenv("LORA_R", "12") | |
| profile = apply_profile_env_overrides(resolve_model_profile("qwen3_4b_2507")) | |
| assert profile.base_model_id == "example/base" | |
| assert profile.hf_model_repo_id == "example/adapter" | |
| assert profile.lora_r == 12 | |