Spaces:
Running
on
Zero
Running
on
Zero
kunhaokhliu
commited on
Commit
·
7ab2a0b
1
Parent(s):
b0fa3d9
update
Browse files- app.py +1 -1
- build.sh +1 -0
- requirements.txt +1 -2
app.py
CHANGED
|
@@ -7,7 +7,7 @@ import torch
|
|
| 7 |
from torchvision.io import write_video
|
| 8 |
from omegaconf import OmegaConf
|
| 9 |
from einops import rearrange
|
| 10 |
-
import
|
| 11 |
|
| 12 |
from pipeline import CausalInferencePipeline
|
| 13 |
from huggingface_hub import snapshot_download, hf_hub_download
|
|
|
|
| 7 |
from torchvision.io import write_video
|
| 8 |
from omegaconf import OmegaConf
|
| 9 |
from einops import rearrange
|
| 10 |
+
import gradio as gr
|
| 11 |
|
| 12 |
from pipeline import CausalInferencePipeline
|
| 13 |
from huggingface_hub import snapshot_download, hf_hub_download
|
build.sh
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
pip install flash-attn --no-build-isolation
|
requirements.txt
CHANGED
|
@@ -40,5 +40,4 @@ flask-socketio
|
|
| 40 |
torchao
|
| 41 |
tensorboard
|
| 42 |
ninja
|
| 43 |
-
packaging
|
| 44 |
-
flash-attn
|
|
|
|
| 40 |
torchao
|
| 41 |
tensorboard
|
| 42 |
ninja
|
| 43 |
+
packaging
|
|
|