Spaces:

chenjoya
/

LiveCC

Running on Zero

chenjoya commited on Apr 23

Commit

ea5bc09

verified ·

1 Parent(s): 8d85e56

Update demo/infer.py

Files changed (1) hide show

demo/infer.py CHANGED Viewed

@@ -160,10 +160,12 @@ class LiveCCDemoInfer:
     @torch.inference_mode()
     def video_qa(
         self,
-        query: str,
         state: dict,
         do_sample: bool = False,
         repetition_penalty: float = 1.05,
         **kwargs,
     ):
         """

     @torch.inference_mode()
     def video_qa(
         self,
+        message: str,
+        history: list,
         state: dict,
         do_sample: bool = False,
         repetition_penalty: float = 1.05,
+        hf_space: bool = False,
         **kwargs,
     ):
         """