khang119966 commited on
Commit
41f100f
·
verified ·
1 Parent(s): 70800b1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -0
app.py CHANGED
@@ -35,6 +35,14 @@ from concurrent.futures import ProcessPoolExecutor
35
 
36
  subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
37
 
 
 
 
 
 
 
 
 
38
  torch.set_default_device('cuda')
39
 
40
 
 
35
 
36
  subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
37
 
38
+ subprocess.run('snap install chromium', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
39
+
40
+ subprocess.run('wget https://dl.google.com/linux/direct/google-chrome-stable_current_amd64.deb', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
41
+
42
+ subprocess.run('dpkg -i google-chrome-stable_current_amd64.deb', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
43
+
44
+
45
+
46
  torch.set_default_device('cuda')
47
 
48