Add model name STRING output to UnetLoaderGGUF

[KasumiFukai]

I would like to request a small usability improvement for UnetLoaderGGUF.

Currently, the node lets the user select unet_name, but it only outputs MODEL.
For workflows that save images with metadata, especially JPEG/WebP metadata for Civitai-style upload forms, it is useful to pass the selected GGUF model name to save nodes automatically.

Use cases:

• Save the selected GGUF model name into image metadata
• Use the model name in filenames or output folders
• Avoid manually typing the model name every time the model is changed
• Keep metadata synchronized with the actual loaded model

Suggested change:

RETURN_TYPES = ("MODEL", "STRING")
RETURN_NAMES = ("model", "modelname")

[roobbb]

Hi.

I also think this might be an useful feature for many reasons.
Therefore I changed the local file "../custom_nodes/ComfyUI-GGUF/nodes.py" as a workaround. So I just copied the existing class "UnetLoaderGGUF" and pasted it in as a new class "UnetLoaderGGUFwithName" and changed some things of that "new" class:

class UnetLoaderGGUFwithName:
    @classmethod
    def INPUT_TYPES(s):
        unet_names = [x for x in folder_paths.get_filename_list("unet_gguf")]
        return {
            "required": {
                "unet_name": (unet_names,),
            }
        }

    RETURN_TYPES = ("MODEL", "STRING")
    RETURN_NAMES = ("model", "filename")
    FUNCTION = "load_unet"
    CATEGORY = "bootleg"
    TITLE = "Unet Loader (GGUF) with Name"

    def load_unet(self, unet_name, dequant_dtype=None, patch_dtype=None, patch_on_device=None):
        ops = GGMLOps()

        if dequant_dtype in ("default", None):
            ops.Linear.dequant_dtype = None
        elif dequant_dtype in ["target"]:
            ops.Linear.dequant_dtype = dequant_dtype
        else:
            ops.Linear.dequant_dtype = getattr(torch, dequant_dtype)

        if patch_dtype in ("default", None):
            ops.Linear.patch_dtype = None
        elif patch_dtype in ["target"]:
            ops.Linear.patch_dtype = patch_dtype
        else:
            ops.Linear.patch_dtype = getattr(torch, patch_dtype)

        # init model
        unet_path = folder_paths.get_full_path("unet", unet_name)
        sd, extra = gguf_sd_loader(unet_path)

        kwargs = {}
        valid_params = inspect.signature(comfy.sd.load_diffusion_model_state_dict).parameters
        if "metadata" in valid_params:
            kwargs["metadata"] = extra.get("metadata", {})

        model = comfy.sd.load_diffusion_model_state_dict(
            sd, model_options={"custom_operations": ops}, **kwargs,
        )
        if model is None:
            logging.error("ERROR UNSUPPORTED UNET {}".format(unet_path))
            raise RuntimeError("ERROR: Could not detect model type of: {}".format(unet_path))
        model = GGUFModelPatcher.clone(model)
        model.patch_on_device = patch_on_device
        return (model, unet_name)

Finally I added a new line on the bottom of the nodey.py file to extend the mappings:

NODE_CLASS_MAPPINGS = {
    "UnetLoaderGGUF": UnetLoaderGGUF,
    "UnetLoaderGGUFwithName": UnetLoaderGGUFwithName,
    "CLIPLoaderGGUF": CLIPLoaderGGUF,
    "DualCLIPLoaderGGUF": DualCLIPLoaderGGUF,
    "TripleCLIPLoaderGGUF": TripleCLIPLoaderGGUF,
    "QuadrupleCLIPLoaderGGUF": QuadrupleCLIPLoaderGGUF,
    "UnetLoaderGGUFAdvanced": UnetLoaderGGUFAdvanced,
}

After restarting ComfyUI it appears among the other GGUF nodes and it works fine in my workflows so far. Of course it is not persistent and gets overwritten each time an update happens.
Maybe city96 will have the time to add such feature to his official code/ repo.

best regards
roobbb