Function get_activation

diffusers/src/diffusers/models/activations.py:36–50

Helper function to get activation function from string. Args: act_fn (str): Name of activation function. Returns: nn.Module: Activation function.

(act_fn: str)

def get_activation(act_fn: str) -> nn.Module:
    """Helper function to get activation function from string.

    Args:
        act_fn (str): Name of activation function.

    Returns:
        nn.Module: Activation function.
    """

    act_fn = act_fn.lower()
    if act_fn in ACTIVATION_FUNCTIONS:
        return ACTIVATION_FUNCTIONS[act_fn]
    else:
        raise ValueError(f"Unsupported activation function: {act_fn}")


class FP32SiLU(nn.Module):