justinj92 commited on
Commit
da076c6
1 Parent(s): 213dfb2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -6,8 +6,9 @@ import torch
6
  import spaces
7
  import subprocess
8
 
 
 
9
  subprocess.run(
10
- "pip install git+https://github.com/huggingface/transformers/",
11
  "pip install flash-attn --no-build-isolation",
12
  env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
13
  shell=True,
 
6
  import spaces
7
  import subprocess
8
 
9
+ subprocess.check_call([sys.executable, '-m', 'pip', 'install', 'git+https://github.com/huggingface/transformers', '-U'])
10
+
11
  subprocess.run(
 
12
  "pip install flash-attn --no-build-isolation",
13
  env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
14
  shell=True,