avoid unnecessary padding

amazon-science · Mar 24, 2024 · afbd3f7 · afbd3f7
1 parent 73be250
commit afbd3f7
Showing 1 changed file with 0 additions and 7 deletions.
diff --git a/src/chronos/chronos.py b/src/chronos/chronos.py
@@ -139,13 +139,6 @@ def input_transform(
 
         if length > self.config.context_length:
             context = context[..., -self.config.context_length :]
-        elif length < self.config.context_length:
-            padding_size = (
-                *context.shape[:-1],
-                self.config.context_length - length,
-            )
-            padding = torch.full(size=padding_size, fill_value=torch.nan)
-            context = torch.concat((padding, context), dim=-1)
 
         attention_mask = ~torch.isnan(context)
         scale = torch.nansum(