fix(loaders): correct random checkpoint loader return type annotation

load_checkpoint returns a 4-tuple (MODEL, CLIP, VAE, model_name) since the
random loader exposes the selected model name; the annotation still claimed
a 3-tuple.
This commit is contained in:
Will Miao
2026-08-15 08:51:39 +08:00
parent 795036275a
commit ae185ee714
+1 -1
View File
@@ -171,7 +171,7 @@ class RandomCheckpointLoaderLM:
ckpt_name: str,
select_at_random: bool = False,
base_model: str = "Any",
) -> Tuple[Any, Any, Any]:
) -> Tuple[Any, Any, Any, str]:
"""Load a checkpoint by name, supporting extra folder paths
Args: