PommesPeter commited on
Commit
b91cd97
1 Parent(s): 95eaf4d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import subprocess
2
  subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
3
 
@@ -12,7 +13,6 @@ import random
12
  import socket
13
  import spaces
14
  import traceback
15
- import os
16
 
17
  import fairscale.nn.model_parallel.initialize as fs_init
18
  import gradio as gr
 
1
+ import os
2
  import subprocess
3
  subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
4
 
 
13
  import socket
14
  import spaces
15
  import traceback
 
16
 
17
  import fairscale.nn.model_parallel.initialize as fs_init
18
  import gradio as gr