Update ts_generation_mixin.py

ts_generation_mixin.py

@@ -39,7 +39,7 @@ class FalconTSTGenerationMixin(GenerationMixin):
         channel = 1
         if len(inputs.shape) == 3:
             channel = inputs.shape[2]
-            inputs = inputs.transpose(0, 2, 1).reshape(batch_size * channel, length)
+            inputs = inputs.permute(0, 2, 1).reshape(batch_size * channel, length)
         elif len(inputs.shape) > 3:
             raise ValueError("Input shape must be [batch, seq_len, channel] or [batch, seq_len]")