chenjoya commited on
Commit
924633d
·
verified ·
1 Parent(s): c8f253e

Update demo/infer.py

Browse files
Files changed (1) hide show
  1. demo/infer.py +1 -1
demo/infer.py CHANGED
@@ -32,7 +32,7 @@ class LiveCCDemoInfer:
32
  streaming_time_interval = streaming_fps_frames / fps
33
  frame_time_interval = 1 / fps
34
 
35
- def __init__(self, model_path: str = None, device: str = 'cpu'):
36
  self.model = Qwen2VLForConditionalGeneration.from_pretrained(
37
  model_path, torch_dtype="auto",
38
  device_map=device,
 
32
  streaming_time_interval = streaming_fps_frames / fps
33
  frame_time_interval = 1 / fps
34
 
35
+ def __init__(self, model_path: str = None, device: str = 'cuda'):
36
  self.model = Qwen2VLForConditionalGeneration.from_pretrained(
37
  model_path, torch_dtype="auto",
38
  device_map=device,