code: def __init__(self, mlp_extra_grouping=True, mp_size=1):
      self.dense_scales = []
      self.qkv_scales = []
      self.mlp4hh_scales = []
      self.mlph4h_scales = []
      self.mlp_extra_grouping = mlp_extra_grouping
      self.mp_size = mp_size
  
file_path: H2O/h2o_flexgen/benchmark/third_party/DeepSpeed/deepspeed/runtime/weight_quantizer.py
imports: import torch
  from ..module_inject.replace_policy import HFBertLayerPolicy, replace_policies
label_desc: Use image node_id 1 to create a new WeightQuantization object from inherited base classes: object with example: obj = WeightQuantization(mlp_extra_grouping, mp_size)