Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
449f481
1
Parent(s):
76d76d6
Install flash-attn without build isolation in app.py and remove it from requirements.txt
Browse files- app.py +3 -0
- 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
|