Buckets:

HuggingFaceDocBuilder's picture
|
download
raw
1.95 kB

Activation functions

Customized activation functions for supporting various models in 🤗 Diffusers.

GELU[[diffusers.models.activations.GELU]]

  • dim_in (int) -- The number of channels in the input.
  • dim_out (int) -- The number of channels in the output.
  • approximate (str, optional, defaults to "none") -- If "tanh", use tanh approximation.
  • bias (bool, defaults to True) -- Whether to use a bias in the linear layer.

GELU activation function with tanh approximation support with approximate="tanh".

GEGLU[[diffusers.models.activations.GEGLU]]

  • dim_in (int) -- The number of channels in the input.
  • dim_out (int) -- The number of channels in the output.
  • bias (bool, defaults to True) -- Whether to use a bias in the linear layer.

A variant of the gated linear unit activation function.

ApproximateGELU[[diffusers.models.activations.ApproximateGELU]]

  • dim_in (int) -- The number of channels in the input.
  • dim_out (int) -- The number of channels in the output.
  • bias (bool, defaults to True) -- Whether to use a bias in the linear layer.

The approximate form of the Gaussian Error Linear Unit (GELU). For more details, see section 2 of this paper.

SwiGLU[[diffusers.models.activations.SwiGLU]]

  • dim_in (int) -- The number of channels in the input.
  • dim_out (int) -- The number of channels in the output.
  • bias (bool, defaults to True) -- Whether to use a bias in the linear layer.

A variant of the gated linear unit activation function. It's similar to GEGLU but uses SiLU / Swish instead of GeLU.

FP32SiLU[[diffusers.models.activations.FP32SiLU]]

SiLU activation function with input upcasted to torch.float32.

LinearActivation[[diffusers.models.activations.LinearActivation]]

Xet Storage Details

Size:
1.95 kB
·
Xet hash:
1ba18307cc6f7f9c5734f0dc360bbcc59903ec9bd16ea45813c0cb39f959d1dd

Xet efficiently stores files, intelligently splitting them into unique chunks and accelerating uploads and downloads. More info.