class_docstr: Base mixin for unmebedding modules. The interface represented by this module is:
  
  if config.do_normalization
      hidden = layer_norm(hidden)
  logits = hidden @ projection
class_docstr_tok: ["Base","mixin","for","unmebedding","modules",".","The","interface","represented","by","this","module","is",":","if","config.do_normalization","hidden","=","layer_norm","(","hidden",")","logits","=","hidden","@","projection"]
code: def config_class() -> Type[DeepSpeedConfigModel]:
      return DSUnembedConfig
  
file_path: DeepSpeed/deepspeed/inference/v2/modules/interfaces/unembed_base.py
imports: from typing import Any, Dict, Optional, Type
  import torch
  from deepspeed.runtime.config_utils import DeepSpeedConfigModel
  from ...ragged import RaggedBatchWrapper
  from ..ds_module import DSModuleBase
  from ..module_registry import DSModuleRegistryBase
  from ..configs import DSUnembedConfig
label_desc: Use image node_id 1 for calling the DSUnembedBase obj's underlying member method code with example usage: obj.config_class() and returns: DSUnembedConfig