lamhieu commited on
Commit
88ac392
·
1 Parent(s): 2b03ec9

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,17 +1,17 @@
1
  # pylint: skip-file
2
 
3
- import subprocess
4
  import json
5
  import requests
6
  import zlib
7
  import random
8
  from PIL import Image
9
 
10
- subprocess.run(
11
- f"pip install flash-attn --no-build-isolation",
12
- env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
13
- shell=True,
14
- )
15
 
16
  import os
17
  from threading import Thread
 
1
  # pylint: skip-file
2
 
3
+ # import subprocess
4
  import json
5
  import requests
6
  import zlib
7
  import random
8
  from PIL import Image
9
 
10
+ # subprocess.run(
11
+ # f"pip install flash-attn --no-build-isolation",
12
+ # env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
13
+ # shell=True,
14
+ # )
15
 
16
  import os
17
  from threading import Thread