diff --git a/configs/vae_magvit_v2/inference/16x128x128.py b/configs/vae_magvit_v2/inference/16x128x128.py
index 5d25983..28b8f01 100644
--- a/configs/vae_magvit_v2/inference/16x128x128.py
+++ b/configs/vae_magvit_v2/inference/16x128x128.py
@@ -7,6 +7,7 @@ is_vae = True
 # Define dataset
 root = None
 data_path = "CSV_PATH"
+max_test_samples = -1
 use_image_transform = False
 num_workers = 4
 video_contains_first_frame = False
diff --git a/configs/vae_magvit_v2/inference/pipeline_16x128x128.py b/configs/vae_magvit_v2/inference/pipeline_16x128x128.py
index a67ea55..24ad8f6 100644
--- a/configs/vae_magvit_v2/inference/pipeline_16x128x128.py
+++ b/configs/vae_magvit_v2/inference/pipeline_16x128x128.py
@@ -8,6 +8,7 @@ use_pipeline = True
 # Define dataset
 root = None
 data_path = "CSV_PATH"
+max_test_samples = -1
 use_image_transform = False
 num_workers = 4
 video_contains_first_frame = False