Spaces:
Running
on
Zero
Running
on
Zero
Update demo/infer.py
Browse files- demo/infer.py +3 -1
demo/infer.py
CHANGED
@@ -160,10 +160,12 @@ class LiveCCDemoInfer:
|
|
160 |
@torch.inference_mode()
|
161 |
def video_qa(
|
162 |
self,
|
163 |
-
|
|
|
164 |
state: dict,
|
165 |
do_sample: bool = False,
|
166 |
repetition_penalty: float = 1.05,
|
|
|
167 |
**kwargs,
|
168 |
):
|
169 |
"""
|
|
|
160 |
@torch.inference_mode()
|
161 |
def video_qa(
|
162 |
self,
|
163 |
+
message: str,
|
164 |
+
history: list,
|
165 |
state: dict,
|
166 |
do_sample: bool = False,
|
167 |
repetition_penalty: float = 1.05,
|
168 |
+
hf_space: bool = False,
|
169 |
**kwargs,
|
170 |
):
|
171 |
"""
|