core.ssm.gdn_layer_config#

Module Contents#

Classes#

GDNLayerConfig

Configuration for a Gated DeltaNet layer in a hybrid stack.

API#

class core.ssm.gdn_layer_config.GDNLayerConfig#

Bases: megatron.core.transformer.transformer_config.TransformerConfig

Configuration for a Gated DeltaNet layer in a hybrid stack.

Due to backwards-compatibility, this config’s arguments are defined in TransformerConfig.