lamhieu commited on
Commit
1cca60b
·
1 Parent(s): fc4a623

chore: update something

Browse files
Files changed (2) hide show
  1. README.md +1 -1
  2. app.py +6 -6
README.md CHANGED
@@ -7,7 +7,7 @@ colorTo: blue
7
  sdk: gradio
8
  sdk_version: 4.36.1
9
  app_file: app.py
10
- pinned: true
11
  header: mini
12
  suggested_hardware: a10g-small
13
  language:
 
7
  sdk: gradio
8
  sdk_version: 4.36.1
9
  app_file: app.py
10
+ pinned: false
11
  header: mini
12
  suggested_hardware: a10g-small
13
  language:
app.py CHANGED
@@ -1,16 +1,16 @@
1
  # pylint: skip-file
2
 
3
- import subprocess
4
  import json
5
  import requests
6
  import zlib
7
  from PIL import Image
8
 
9
- subprocess.run(
10
- f"pip install flash-attn --no-build-isolation",
11
- env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
12
- shell=True,
13
- )
14
 
15
  import os
16
  from threading import Thread
 
1
  # pylint: skip-file
2
 
3
+ # import subprocess
4
  import json
5
  import requests
6
  import zlib
7
  from PIL import Image
8
 
9
+ # subprocess.run(
10
+ # f"pip install flash-attn --no-build-isolation",
11
+ # env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
12
+ # shell=True,
13
+ # )
14
 
15
  import os
16
  from threading import Thread