O S I H commited on
Commit
79867a4
1 Parent(s): abfec2c

reinstall transformers

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -11,12 +11,17 @@ import spaces
11
  import time
12
  import subprocess
13
 
 
 
 
 
14
  subprocess.run(
15
  "pip install flash-attn --no-build-isolation",
16
  env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
17
  shell=True,
18
  )
19
 
 
20
  token = os.environ["HF_TOKEN"]
21
 
22
 
 
11
  import time
12
  import subprocess
13
 
14
+ subprocess.run(
15
+ "pip uninstall -y transformers && pip install git+https://github.com/huggingface/transformers",
16
+ shell=True,
17
+ )
18
  subprocess.run(
19
  "pip install flash-attn --no-build-isolation",
20
  env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
21
  shell=True,
22
  )
23
 
24
+
25
  token = os.environ["HF_TOKEN"]
26
 
27