Ivy1997 commited on
Commit
ecb1b17
·
verified ·
1 Parent(s): 59a8fda

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -1,10 +1,10 @@
1
  import gradio as gr
2
  import subprocess
3
- subprocess.run(
4
- "pip install flash-attn --no-build-isolation",
5
- env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
6
- shell=True,
7
- )
8
  from llava.model.builder import load_pretrained_model
9
  from llava.mm_utils import process_images, tokenizer_image_token
10
  from llava.constants import IMAGE_TOKEN_INDEX, DEFAULT_IMAGE_TOKEN
 
1
  import gradio as gr
2
  import subprocess
3
+ # subprocess.run(
4
+ # "pip install flash-attn --no-build-isolation",
5
+ # env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
6
+ # shell=True,
7
+ # )
8
  from llava.model.builder import load_pretrained_model
9
  from llava.mm_utils import process_images, tokenizer_image_token
10
  from llava.constants import IMAGE_TOKEN_INDEX, DEFAULT_IMAGE_TOKEN