Hugging Face's logo

Spaces:
HuggingFaceM4
/
idefics-8b
Running on Zero

App Files Community
1
merve HF staff commited on
Commit
8e85e31
•
1 Parent(s): 5cb8d24

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -7,6 +7,9 @@ import time
7
  from PIL import Image
8
  import torch
9
  import spaces
 
 
 
10
 
11
  PROCESSOR = AutoProcessor.from_pretrained("HuggingFaceM4/idefics2-8b")
12
 
 
7
  from PIL import Image
8
  import torch
9
  import spaces
10
+ import subprocess
11
+ subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
12
+
13
 
14
  PROCESSOR = AutoProcessor.from_pretrained("HuggingFaceM4/idefics2-8b")
15