remove leftover mid references (#491)

2026-02-02 17:33:46 +01:00
parent b19b4f3e49
commit 72b9064f9d
7 changed files with 6 additions and 12 deletions
@@ -164,7 +164,6 @@ def load_model_from_dir(checkpoints_dir, device, phase, model_tag=None, step=Non
 def load_model(source, *args, **kwargs):
    model_dir = {
        "base": "base_checkpoints",
-        "mid": "mid_checkpoints",
        "sft": "chatsft_checkpoints",
        "rl": "chatrl_checkpoints",
    }[source]