Aliayub1995
commited on
Commit
•
c046ba1
1
Parent(s):
89a6901
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:
|