From 1facd14b1da33ab7aa27a982b9953b5d201fe720 Mon Sep 17 00:00:00 2001
From: USTC-liuchang <68578066+AlonzoLeeeooo@users.noreply.github.com>
Date: Sat, 1 Jun 2024 09:59:57 +0800
Subject: [PATCH] Update dataset.py

Fix critical error in `dataset.py`, which might cause dead loop while running the code.
---
 animatediff/data/dataset.py | 9 +--------
 1 file changed, 1 insertion(+), 8 deletions(-)

diff --git a/animatediff/data/dataset.py b/animatediff/data/dataset.py
index 3f6ec102..d6ea18d6 100644
--- a/animatediff/data/dataset.py
+++ b/animatediff/data/dataset.py
@@ -64,14 +64,7 @@ def __len__(self):
         return self.length
 
     def __getitem__(self, idx):
-        while True:
-            try:
-                pixel_values, name = self.get_batch(idx)
-                break
-
-            except Exception as e:
-                idx = random.randint(0, self.length-1)
-
+        pixel_values, name = self.get_batch(idx)
         pixel_values = self.pixel_transforms(pixel_values)
         sample = dict(pixel_values=pixel_values, text=name)
         return sample