Update modeling_mists.py

modeling_mists.py

@@ -239,6 +239,13 @@ class MistsForConditionalGeneration(MistsPreTrainedModel):
         #     else self.config.vision_feature_select_strategy
         # )
 
+        print("model inputs:")
+        print("input_ids: ", input_ids.shape if input_ids is not None else None)
+        print("attention_mask: ", attention_mask.shape if attention_mask is not None else None)
+        print("time_series_values: ", time_series_values.shape if time_series_values is not None else None)
+        print("time_series_input_mask: ", time_series_input_mask.shape if time_series_input_mask is not None else None)
+        print("==========")
+
         if inputs_embeds is None:
             # 1. Extra the input embeddings
             inputs_embeds = self.get_input_embeddings()(input_ids)