SuleBai commited on
Commit
b013151
·
1 Parent(s): ca8c3e9
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -11,7 +11,6 @@ model_path = 'GD-ML/UniVG-R1'
11
  model = Qwen2VLForConditionalGeneration.from_pretrained(
12
  model_path,
13
  torch_dtype=torch.bfloat16,
14
- attn_implementation="flash_attention_2",
15
  device_map="cuda:0",
16
  )
17
  processor = AutoProcessor.from_pretrained(model_path, max_pixels=401408)
 
11
  model = Qwen2VLForConditionalGeneration.from_pretrained(
12
  model_path,
13
  torch_dtype=torch.bfloat16,
 
14
  device_map="cuda:0",
15
  )
16
  processor = AutoProcessor.from_pretrained(model_path, max_pixels=401408)