MohamedRashad commited on
Commit
adbfbf9
·
verified ·
1 Parent(s): 837a765

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -1,3 +1,6 @@
 
 
 
1
  from transformers import (
2
  NougatProcessor,
3
  VisionEncoderDecoderModel,
 
1
+ import os
2
+ os.system("pip install -U flash-attn")
3
+
4
  from transformers import (
5
  NougatProcessor,
6
  VisionEncoderDecoderModel,