defined self.nframe as used in the generate_inner

open-compass · Jan 6, 2025 · f0b7e2a · f0b7e2a
1 parent b96a00e
commit f0b7e2a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/vlmeval/vlm/video_llm/video_llava.py b/vlmeval/vlm/video_llm/video_llava.py
@@ -44,6 +44,7 @@ def __init__(self, model_path='LanguageBind/Video-LLaVA-7B-hf', **kwargs):
         self.model.eval().cuda()
         self.processor = VideoLlavaProcessor.from_pretrained(model_path)
         self.kwargs = kwargs
+        self.nframe = 8
         torch.cuda.empty_cache()
 
     def generate_inner(self, message, dataset=None):