chunk_content:  + sample_input_ids
          model_inputs["attention_mask"][i] = [0] * (max_length - len(sample_input_ids)) + model_inputs[
              "attention_mask"
          ][i]
          labels["input_ids"][i] = [-100] * (max_length - len(sample_input_ids)) + label_input_ids
          model_inputs["input_ids"][i] = torch.tensor(model_inputs["input_ids"][i][:max_length])
          model_inputs["attention_mask"][i] = torch.tensor(model_inputs["attention_mask"][i][