prithivMLmods commited on
Commit
9632ea7
·
verified ·
1 Parent(s): 5271c91

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -10
app.py CHANGED
@@ -32,16 +32,6 @@ from transformers.image_utils import load_image
32
  from gradio.themes import Soft
33
  from gradio.themes.utils import colors, fonts, sizes
34
 
35
-
36
- #Attempt to install flash-attn
37
- try:
38
- subprocess.run('pip install flash_attn==2.3.4 --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, check=True, shell=True)
39
- except subprocess.CalledProcessError as e:
40
- print(f"Error installing flash-attn: {e}")
41
- print("Continuing without flash-attn.")
42
-
43
-
44
-
45
  colors.steel_blue = colors.Color(
46
  name="steel_blue",
47
  c50="#EBF3F8",
 
32
  from gradio.themes import Soft
33
  from gradio.themes.utils import colors, fonts, sizes
34
 
 
 
 
 
 
 
 
 
 
 
35
  colors.steel_blue = colors.Color(
36
  name="steel_blue",
37
  c50="#EBF3F8",