chenjoya commited on
Commit
df85f32
·
1 Parent(s): 8bd79af
Files changed (1) hide show
  1. 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
- # attn_implementation='sdpa'
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)