Function create_adapter

espnet2/layers/create_adapter.py:23–40

Create adapter for the base model. Args: model (torch.nn.Module): Base model to be adapted. adapter_type (str): Name of adapter adapter_conf (dict): Configuration for the adapter e.g. {"rank": 8, "alpha": 8, ...} for lora

(
    model: torch.nn.Module,
    adapter: str,
    adapter_conf: dict,
)

@typechecked
def create_adapter(
    model: torch.nn.Module,
    adapter: str,
    adapter_conf: dict,
):
    """Create adapter for the base model.


    Args:
        model (torch.nn.Module): Base model to be adapted.
        adapter_type (str): Name of adapter
        adapter_conf (dict): Configuration for the adapter
            e.g.  {"rank": 8, "alpha": 8, ...} for lora

    """
    assert adapter in create_adapter_fn_table, f"Adapter {adapter} is not supported."
    create_adapter_fn = create_adapter_fn_table[adapter]
    create_adapter_fn(model=model, **adapter_conf)