diff --git a/cotracker/models/evaluation_predictor.py b/cotracker/models/evaluation_predictor.py
index aa7308d..87f8e18 100644
--- a/cotracker/models/evaluation_predictor.py
+++ b/cotracker/models/evaluation_predictor.py
@@ -38,7 +38,6 @@ class EvaluationPredictor(torch.nn.Module):
         B, N, D = queries.shape
 
         assert D == 3
-        assert B == 1
 
         video = video.reshape(B * T, C, H, W)
         video = F.interpolate(video, tuple(self.interp_shape), mode="bilinear", align_corners=True)
diff --git a/cotracker/predictor.py b/cotracker/predictor.py
index 4122f0b..dc2cd96 100644
--- a/cotracker/predictor.py
+++ b/cotracker/predictor.py
@@ -92,7 +92,6 @@ class CoTrackerPredictor(torch.nn.Module):
         backward_tracking=False,
     ):
         B, T, C, H, W = video.shape
-        assert B == 1
 
         video = video.reshape(B * T, C, H, W)
         video = F.interpolate(video, tuple(self.interp_shape), mode="bilinear", align_corners=True)