code: def __init__(self, model, tokenizer, prompt_length=32):
      model.eval()
      self.model = model
      self.tokenizer = tokenizer
      self.prompt_length = prompt_length
  
file_path: DeepKE/example/llm/CPM-Bee/src/cpm_live/generation/ant.py
imports: import torch
  import torch.nn.functional
  from .generation_utils import BeamHypotheses, apply_repetition_penalty, top_k_top_p_filtering
  from ..utils import pad
label_desc: Use image node_id 1 to create a new CPMAntGeneration object  with example: obj = CPMAntGeneration(model, tokenizer, prompt_length)