fix: adjust trtllm looper for video chunk enum

huggingface · Dec 16, 2024 · 4e921bf · 4e921bf
1 parent ffc2656
commit 4e921bf
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/backends/trtllm/src/looper.rs b/backends/trtllm/src/looper.rs
@@ -301,6 +301,7 @@ impl TensorRtLlmBackendV2 {
             1 => match request.inputs.first().expect("Single item-chunk") {
                 Chunk::Text(_) => Ok(()),
                 Chunk::Image(_) => Err(ValidationError(UnsupportedModality("image"))),
+                Chunk::Video(_) => Err(ValidationError(UnsupportedModality("video"))),
             },
         }
     }