Update videollama2/__init__.py
Browse files- videollama2/__init__.py +1 -1
videollama2/__init__.py
CHANGED
|
@@ -33,7 +33,7 @@ def model_init(model_path=None, **kwargs):
|
|
| 33 |
return model, processor, tokenizer
|
| 34 |
|
| 35 |
|
| 36 |
-
def mm_infer(
|
| 37 |
"""inference api of VideoLLaMA2 for video understanding.
|
| 38 |
|
| 39 |
Args:
|
|
|
|
| 33 |
return model, processor, tokenizer
|
| 34 |
|
| 35 |
|
| 36 |
+
def mm_infer(image_or_video, instruct, model, tokenizer, modal='video', **kwargs):
|
| 37 |
"""inference api of VideoLLaMA2 for video understanding.
|
| 38 |
|
| 39 |
Args:
|