Fix stopping criteria window size calculation to prevent index errors

modeling_emuru.py

@@ -252,7 +252,7 @@ class Emuru(PreTrainedModel):
 
             if stopping_criteria == 'latent':
                 similarity = torch.nn.functional.cosine_similarity(canvas_sequence, pad_token, dim=-1)
-                windows = (similarity > self.padding_token_threshold).unfold(1, stopping_after, 1)
+                windows = (similarity > self.padding_token_threshold).unfold(1, min(stopping_after, similarity.size(-1)), 1)
                 window_sums = windows.to(torch.int).sum(dim=2)
 
                 for i in range(similarity.size(0)):