BK-Lee commited on
Commit
a5af071
1 Parent(s): b6b8ad8
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -16,7 +16,8 @@ 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', '-U'])
 
20
 
21
  # loading meteor model
22
  mmamba = load_mmamba('BK-Lee/Meteor-Mamba').to('cuda')
 
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
 
22
  # loading meteor model
23
  mmamba = load_mmamba('BK-Lee/Meteor-Mamba').to('cuda')