Fix temporal chunking to use H3 token grid

2026-09-04 07:06:08 +00:00
parent 04a61af874
commit 782a7d658b
2 changed files with 63 additions and 22 deletions
@@ -1181,8 +1181,10 @@ class DumasH3LongVideosHelperTests(unittest.TestCase):

    def test_temporal_segments_split_long_sequences(self):
        latent = importlib.import_module("dumas_h3_latent_upscale")
-        bounds = latent._temporal_segments(124, 85, 17)
-        self.assertEqual(bounds, [(0, 85), (68, 124)])
+        bounds = latent._temporal_segments(36, 85, 17)
+        self.assertGreater(len(bounds), 1)
+        self.assertEqual(bounds[0][0], 0)
+        self.assertEqual(bounds[-1][2], 36)

    def test_shrink_temporal_param_reduces_chunk_length(self):
        latent = importlib.import_module("dumas_h3_latent_upscale")