Update video_processing_moss_vl.py

video_processing_moss_vl.py

@@ -231,7 +231,17 @@ def split_indices(indices: List[Union[int, float]], num_chunks: int) -> List[Lis
         
     Returns:
         List of index chunks.
+
+    Raises:
+        ValueError: If indices is empty or num_chunks is not positive.
     """
+    if len(indices) == 0:
+        raise ValueError("indices must not be empty")
+    if num_chunks <= 0:
+        raise ValueError("num_chunks must be positive")
+
+    # Never create empty decode jobs when there are fewer frames than workers.
+    num_chunks = min(num_chunks, len(indices))
     chunk_size = len(indices) // num_chunks
     chunks = []
     for i in range(num_chunks - 1):
@@ -272,7 +282,7 @@ def decode_with_multithreading(indices: List[int], num_threads: int, video_path:
         dict: Contains 'data', 'duration_seconds', 'pts_seconds' tensors.
     """
     chunks = split_indices(indices, num_chunks=num_threads)
-    results = Parallel(n_jobs=num_threads, prefer="threads", verbose=0)(
+    results = Parallel(n_jobs=len(chunks), prefer="threads", verbose=0)(
         delayed(decode_sequentially)(chunk, video_path) for chunk in chunks
     )
     
@@ -319,7 +329,7 @@ def timestamp_decode_with_multithreading(timestamp_list: List[float], num_thread
         dict: Contains 'data', 'duration_seconds', 'pts_seconds' tensors.
     """
     chunks = split_indices(timestamp_list, num_chunks=num_threads)
-    results = Parallel(n_jobs=num_threads, prefer="threads", verbose=0)(
+    results = Parallel(n_jobs=len(chunks), prefer="threads", verbose=0)(
         delayed(decode_sequentially_timestamp)(chunk, video_path) for chunk in chunks
     )
     
@@ -1148,4 +1158,3 @@ class MossVLVideoProcessor(BaseVideoProcessor):
 
 
 __all__ = ["MossVLVideoProcessor"]
-