manu commited on
Commit
8a50866
·
verified ·
1 Parent(s): 7a0eade

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -11,7 +11,8 @@ from tqdm import tqdm
11
 
12
  from colpali_engine.models import ColQwen2, ColQwen2Processor
13
 
14
-
 
15
 
16
  model = ColQwen2.from_pretrained(
17
  "manu/colqwen2-v1.0-alpha",
 
11
 
12
  from colpali_engine.models import ColQwen2, ColQwen2Processor
13
 
14
+ print("Install FA2")
15
+ os.system("pip install flash-attn --no-build-isolation")
16
 
17
  model = ColQwen2.from_pretrained(
18
  "manu/colqwen2-v1.0-alpha",