Update modeling_qwen2.py (#33)

- Update modeling_qwen2.py (38907af6c6838420f3a233eb9cc00833bba592ce)


Co-authored-by: Zhirui Zhang <[email protected]>

modeling_qwen2.py

@@ -174,16 +174,17 @@ class TimeSeriesEmbedding(nn.Module):
                 padding = last_value.repeat(padding_length, 1)
                 xi = torch.cat([xi, padding], dim=0)
                 
-                # Use special padding index for padding positions
-                padding_positions = torch.full((padding_length,), self.padding_idx, device=x.device)
-                position_indices = torch.cat([position_indices, padding_positions], dim=0)
+                # Use special padding index for padding positions when use_position_embedding enabled
+                if self.use_position_embedding:
+                    padding_positions = torch.full((padding_length,), self.padding_idx, device=x.device)
+                    position_indices = torch.cat([position_indices, padding_positions], dim=0)
 
             # Reshape to patches
             xi = xi.reshape(pc, self.patch_size)  # (num_patches, patch_size)
-            position_indices = position_indices.reshape(pc, self.patch_size)  # (num_patches, patch_size)
-
+            
             if self.use_position_embedding:
                 # Collect position indices instead of calling embedding immediately
+                position_indices = position_indices.reshape(pc, self.patch_size)  # (num_patches, patch_size)
                 all_position_indices.append(position_indices)
                 patch_info_list.append({
                     'xi': xi,