Spaces:

prithivMLmods
/

Tiny-VLMs-Lab

Running on Zero

prithivMLmods commited on 15 days ago

Commit

3abca60

verified ·

1 Parent(s): d6645ff

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,6 +13,7 @@ import re
 import base64
 import copy
 from dataclasses import dataclass
 # Vision and ML Libraries
 import fitz  # PyMuPDF

 import base64
 import copy
 from dataclasses import dataclass
+import flash_attn_2_cuda as flash_attn_gpu
 # Vision and ML Libraries
 import fitz  # PyMuPDF