KevinQHLin commited on
Commit
e0c72f4
·
verified ·
1 Parent(s): dc5059e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -55,7 +55,7 @@ model = Qwen2VLForConditionalGeneration.from_pretrained(
55
  destination_folder,
56
  torch_dtype=torch.bfloat16,
57
  # device_map="cpu",
58
- attn_implementation = "flash_attention_2",
59
  )
60
 
61
  # Load the processor
 
55
  destination_folder,
56
  torch_dtype=torch.bfloat16,
57
  # device_map="cpu",
58
+ _attn_implementation = "flash_attention_2",
59
  )
60
 
61
  # Load the processor