Update modeling_mists.py

modeling_mists.py

@@ -239,12 +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("==========")
+        # モデルInput確認用（学習時）
+        # 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