chunk_content: en(model_inputs["input_ids"][i])
      # print(model_inputs)
      for i in range(batch_size):
          sample_input_ids = model_inputs["input_ids"][i]
          label_input_ids = labels["input_ids"][i]
          model_inputs["input_ids"][i] = [tokenizer.pad_token_id] * (
              max_length - len(sample_input_ids)
          ) + sample_input_ids
          model_inputs["attention_mask"][i] = [0] * (max_length - len(sample_input_ids)) + model_inputs[
       