MohamedRashad commited on
Commit
449f481
·
1 Parent(s): 76d76d6

Install flash-attn without build isolation in app.py and remove it from requirements.txt

Browse files
Files changed (2) hide show
  1. app.py +3 -0
  2. requirements.txt +0 -1
app.py CHANGED
@@ -1,3 +1,6 @@
 
 
 
1
  import gradio as gr
2
  import spaces
3
  from gradio_litmodel3d import LitModel3D
 
1
+ import subprocess
2
+ subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
3
+
4
  import gradio as gr
5
  import spaces
6
  from gradio_litmodel3d import LitModel3D
requirements.txt CHANGED
@@ -22,6 +22,5 @@ xformers==0.0.27.post2
22
  spconv-cu120
23
  transformers
24
  gradio_litmodel3d
25
- flash-attn
26
  https://huggingface.co/spaces/JeffreyXiang/TRELLIS/resolve/main/wheels/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl?download=true
27
  https://huggingface.co/spaces/JeffreyXiang/TRELLIS/resolve/main/wheels/nvdiffrast-0.3.3-cp310-cp310-linux_x86_64.whl?download=true
 
22
  spconv-cu120
23
  transformers
24
  gradio_litmodel3d
 
25
  https://huggingface.co/spaces/JeffreyXiang/TRELLIS/resolve/main/wheels/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl?download=true
26
  https://huggingface.co/spaces/JeffreyXiang/TRELLIS/resolve/main/wheels/nvdiffrast-0.3.3-cp310-cp310-linux_x86_64.whl?download=true