code/self_healing_finetuning/model_policy/registry.py

from __future__ import annotations

from n21.settings import DEFAULT_MODEL_ID


DEFAULT_CANDIDATES = {
    "model_policy": {
        "default_strategy": "fingpt_first_bootstrap",
        "fallback_strategy": "best_certified_open_weight",
        "selection_order": [
            "approved_fingpt_task_adapter",
            "approved_fingpt_multi_task_model",
            "approved_finance_tuned_open_model",
            "approved_general_open_weight_base",
            "approved_internal_base_model",
        ],
    },
    "model_candidates": [
        {
            "model_id": DEFAULT_MODEL_ID,
            "role": "finance_multitask_lora",
            "track": "fingpt_bootstrap",
            "source_adapter_repo_id": "FinGPT/fingpt-mt_llama3-8b_lora",
            "source_base_model_id": "meta-llama/Meta-Llama-3-8B",
            "license_status": "review_required",
            "provenance_recorded": True,
            "private_deployable": True,
            "continuation_tunable": True,
        },
        {
            "model_id": "openai/gpt-oss-20b",
            "role": "default_open_weight_base_candidate",
            "track": "foundation_track",
            "license_status": "review_required",
            "provenance_recorded": True,
            "private_deployable": True,
            "continuation_tunable": True,
        },
    ],
}