docstr: Necessary to implement for `HybridSplitQKVContainer`
docstr_tok: ["Necessary","to","implement","for","`","HybridSplitQKVContainer","`"]
code: def set_q_k_v(self):
      """
      Necessary to implement for `HybridSplitQKVContainer`
      """
      self.qw = self.policy.client_module.self_attn.q_proj.weight
      self.qb = self.policy.client_module.self_attn.q_proj.bias
      self.kw = self.policy.client_module.self_attn.k_proj.weight
      self.kb = self.policy.client_module.self_attn.k_proj.bias
      self.vw = self.policy.client_module.self_attn.v_proj.weight
      self.vb = self.policy.client_module.self_attn.v_proj.bias
  
imports: import importlib
  import torch
  from torch.nn.parameter import Parameter
  from deepspeed.model_implementations.transformers.ds_gpt import DeepSpeedGPTInference
  from deepspeed.utils.types import ActivationFuncType, NormType
  from ..policy import TransformerPolicy, maybe_copy, maybe_copy_geglu, maybe_copy_qkv, maybe_get_lora, transformer_param_names
  from .base import *
  from .features import HybridGatedMLPContainer, HybridSplitQKVContainer
label_desc: Use image node_id 5 for calling the DS_InternLMContainer obj's underlying member method code with example usage: obj.set_q_k_v() without return types