Spaces:
Runtime error
Runtime error
Update demo/infer.py
Browse files- 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 = '
|
| 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,
|