chunk_content: ut_ids = labels["input_ids"][i] + [tokenizer.pad_token_id]
          # print(i, sample_input_ids, label_input_ids)
          model_inputs["input_ids"][i] = sample_input_ids + label_input_ids
          labels["input_ids"][i] = [-100] * len(sample_input_ids) + label_input_ids
          model_inputs["attention_mask"][i] = [1] * len(model_inputs["input_ids"][i])
      # print(model_inputs)
      for i in range(batch_size):
          sample_input_ids = model_inpu