Update modeling_mists.py

modeling_mists.py

@@ -241,11 +241,11 @@ class MistsForConditionalGeneration(MistsPreTrainedModel):
 
         # モデル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("==========")
+        # 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