code: def get_chunk(lst, n, k):
      chunks = split_list(lst, n)
      return chunks[k]
  
imports: import argparse
  import torch
  import os
  import json
  import cv2
  import numpy
  from icecream import ic
  from tqdm import tqdm
  import shortuuid
  import sys
  from llava.constants import IMAGE_TOKEN_INDEX, DEFAULT_IMAGE_TOKEN, DEFAULT_IM_START_TOKEN, DEFAULT_IM_END_TOKEN
  from llava.conversation import conv_templates, SeparatorStyle
  from llava.model.builder import load_pretrained_model
  from llava.utils import disable_torch_init
  from llava.mm_utils import tokenizer_image_token, get_model_name_from_path, KeywordsStoppingCriteria
  from highlighter_modules.guidance import ProbCFGLogitsProcessor
  from highlighter_modules.utils import txt_highlight_mask
  from highlighter_modules.attention_llama_llava import llava_modify_inf
  from PIL import Image
  import math
label_desc: Use image node_id 2 for calling a global function with example usage: get_chunk(lst, n, k) and returns: chunks