chenjoya commited on
Commit
ea5bc09
·
verified ·
1 Parent(s): 8d85e56

Update demo/infer.py

Browse files
Files changed (1) hide show
  1. 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
- query: str,
 
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
  """