CCSR_Model_Select

Documentation

• Class name: CCSR_Model_Select
• Category: CCSR
• Output node: False

The CCSR_Model_Select node is designed to select and load a specific checkpoint for the CCSR model based on the provided checkpoint name. It facilitates the initialization of the CCSR model with pre-trained weights for further processing or inference tasks.

Input types

Required

• ckpt_name
  • The name of the checkpoint to load. This parameter is crucial for identifying and retrieving the specific pre-trained model weights from the available checkpoints.
  • Comfy dtype: COMBO[STRING]
  • Python dtype: str

Output types

• ccsr_model
  • Comfy dtype: CCSRMODEL
  • The path to the loaded CCSR model checkpoint. This output is essential for subsequent nodes that require the model's checkpoint path for initialization or further processing.
  • Python dtype: str

Usage tips

• Infra type: CPU
• Common nodes: unknown

Source code

class CCSR_Model_Select:
    @classmethod
    def INPUT_TYPES(s):
        return {"required": { 
            "ckpt_name": (folder_paths.get_filename_list("checkpoints"),),                                             
                             }}
    RETURN_TYPES = ("CCSRMODEL",)
    RETURN_NAMES = ("ccsr_model",)
    FUNCTION = "load_ccsr_checkpoint"

    CATEGORY = "CCSR"

    def load_ccsr_checkpoint(self, ckpt_name):
        ckpt_path = folder_paths.get_full_path("checkpoints", ckpt_name)

        return (ckpt_path,)