chunk_content: ext_column])
      inputs = [f"{text_column} : {x} Label : " for x in examples[text_column]]
      targets = [str(x) for x in examples[label_column]]
      model_inputs = tokenizer(inputs)
      labels = tokenizer(targets)
      for i in range(batch_size):
          sample_input_ids = model_inputs["input_ids"][i]
          label_input_ids = labels["input_ids"][i] + [tokenizer.pad_token_id]
          # print(i, sample_input_ids, label_input_ids)
          model_i