Added safety measure

models/vallex.py

@@ -588,6 +588,11 @@ class VALLE(VALLF):
                 print(f"Current memory used: {memory_used:.2f} MB")
                 break
 
+            # safety measure, break if token sequence too long
+            if y.shape[1] > 2250:
+                print(f"VALL-E EOS [{prompts.shape[1]} -> {y.shape[1]}]")
+                break
+
             y = torch.concat([y, samples], dim=1)
 
         codes = [y[:, prefix_len + int(self.ar_audio_prepend_bos) :]]