BK-Lee commited on
Commit
760dfc7
1 Parent(s): 72ccec0
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -16,6 +16,7 @@ from torchvision.transforms.functional import pil_to_tensor
16
  # package install
17
  import sys
18
  import subprocess
 
19
  subprocess.check_call([sys.executable, '-m', 'pip', 'install', 'flash-attn', '--no-build-isolation'])
20
 
21
 
 
16
  # package install
17
  import sys
18
  import subprocess
19
+ subprocess.check_call([sys.executable, '-m', 'pip', 'install', '--upgrade', 'pip'])
20
  subprocess.check_call([sys.executable, '-m', 'pip', 'install', 'flash-attn', '--no-build-isolation'])
21
 
22