Function _add_variant

src/diffusers/utils/hub_utils.py:218–224

(weights_name: str, variant: str | None = None)

def _add_variant(weights_name: str, variant: str | None = None) -> str:
    if variant is not None:
        splits = weights_name.split(".")
        splits = splits[:-1] + [variant] + splits[-1:]
        weights_name = ".".join(splits)

    return weights_name


@validate_hf_hub_args