Spaces:
Running
on
Zero
Running
on
Zero
??
Browse files- demo/infer.py +1 -1
demo/infer.py
CHANGED
@@ -34,7 +34,7 @@ class LiveCCDemoInfer:
|
|
34 |
def __init__(self, model_path: str = None, device_id: int = 0):
|
35 |
self.model = Qwen2VLForConditionalGeneration.from_pretrained(
|
36 |
model_path, torch_dtype="auto",
|
37 |
-
|
38 |
)
|
39 |
self.model.to(f'cuda:{device_id}')
|
40 |
self.processor = AutoProcessor.from_pretrained(model_path, use_fast=False)
|
|
|
34 |
def __init__(self, model_path: str = None, device_id: int = 0):
|
35 |
self.model = Qwen2VLForConditionalGeneration.from_pretrained(
|
36 |
model_path, torch_dtype="auto",
|
37 |
+
attn_implementation='eager'
|
38 |
)
|
39 |
self.model.to(f'cuda:{device_id}')
|
40 |
self.processor = AutoProcessor.from_pretrained(model_path, use_fast=False)
|