Upload modeling_mists.py

modeling_mists.py

@@ -254,11 +254,11 @@ class MistsForConditionalGeneration(MistsPreTrainedModel):
         print("========")
         print("input_ids: ", input_ids.shape)
         print("time_series_values: ", time_series_values.shape)
-        print("inputs_embeds: ", inputs_embeds.shape)
 
         if inputs_embeds is None:
             # 1. Extra the input embeddings
             inputs_embeds = self.get_input_embeddings()(input_ids)
+            print("inputs_embeds: ", inputs_embeds.shape)
 
             # 2. Merge text and time_series
             if time_series_values is not None and input_ids.shape[1] != 1: