Choiszt commited on
Commit
16eee85
·
1 Parent(s): 5b2987d

update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -34,6 +34,7 @@ from egogpt.constants import (
34
  DEFAULT_SPEECH_TOKEN
35
  )
36
  from egogpt.conversation import conv_templates, SeparatorStyle
 
37
  subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
38
 
39
  # pretrained = "/mnt/sfs-common/jkyang/EgoGPT/checkpoints/EgoGPT-llavaov-7b-EgoIT-109k-release"
 
34
  DEFAULT_SPEECH_TOKEN
35
  )
36
  from egogpt.conversation import conv_templates, SeparatorStyle
37
+ import subprocess
38
  subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
39
 
40
  # pretrained = "/mnt/sfs-common/jkyang/EgoGPT/checkpoints/EgoGPT-llavaov-7b-EgoIT-109k-release"