in jat/processing_jat.py [0:0]
def model_input_names(self):
return [
"input_ids",
"attention_mask",
"pixel_values",
"continuous_observations",
"discrete_observations",
"image_observations",
"continuous_actions",
"discrete_actions",
"rewards",
]