From ae185ee7149a5cae5d7459eed76083754712737a Mon Sep 17 00:00:00 2001 From: Will Miao Date: Sat, 15 Aug 2026 08:51:39 +0800 Subject: [PATCH] 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. --- py/nodes/random_checkpoint_loader.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/py/nodes/random_checkpoint_loader.py b/py/nodes/random_checkpoint_loader.py index 5b80b049..b9c4ff24 100644 --- a/py/nodes/random_checkpoint_loader.py +++ b/py/nodes/random_checkpoint_loader.py @@ -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: