class_docstr: Inference model implementation for ragged batching for Llama-2 models.
code: def head_size(self) -> int:
      return self.model_dim // self.n_heads
  
imports: from typing import Iterable, Optional, Tuple
  import torch
  import deepspeed.comm
  from ...allocator import empty_from
  from ...inference_utils import ActivationType, DtypeEnum
  from ..None import *
  from ...modules.configs import *
  from ...modules.interfaces import *
  from ...modules import heuristics
  from ...ragged import RaggedBatchWrapper
  from .container import QwenNonTransformerContainer, QwenTransformerContainer
label_desc: Use image node_id 5 for calling the QwenInferenceModel obj's underlying member method code with example usage: obj.head_size() and returns: unknown