Spaces:
Running
on
L40S
Running
on
L40S
Rex Cheng
commited on
Commit
•
b7f72e1
1
Parent(s):
10fd932
compile
Browse files- .gitattributes +0 -35
- .gitignore +4 -4
- app.py +3 -0
- requirements.txt +1 -1
.gitattributes
DELETED
@@ -1,35 +0,0 @@
|
|
1 |
-
*.7z filter=lfs diff=lfs merge=lfs -text
|
2 |
-
*.arrow filter=lfs diff=lfs merge=lfs -text
|
3 |
-
*.bin filter=lfs diff=lfs merge=lfs -text
|
4 |
-
*.bz2 filter=lfs diff=lfs merge=lfs -text
|
5 |
-
*.ckpt filter=lfs diff=lfs merge=lfs -text
|
6 |
-
*.ftz filter=lfs diff=lfs merge=lfs -text
|
7 |
-
*.gz filter=lfs diff=lfs merge=lfs -text
|
8 |
-
*.h5 filter=lfs diff=lfs merge=lfs -text
|
9 |
-
*.joblib filter=lfs diff=lfs merge=lfs -text
|
10 |
-
*.lfs.* filter=lfs diff=lfs merge=lfs -text
|
11 |
-
*.mlmodel filter=lfs diff=lfs merge=lfs -text
|
12 |
-
*.model filter=lfs diff=lfs merge=lfs -text
|
13 |
-
*.msgpack filter=lfs diff=lfs merge=lfs -text
|
14 |
-
*.npy filter=lfs diff=lfs merge=lfs -text
|
15 |
-
*.npz filter=lfs diff=lfs merge=lfs -text
|
16 |
-
*.onnx filter=lfs diff=lfs merge=lfs -text
|
17 |
-
*.ot filter=lfs diff=lfs merge=lfs -text
|
18 |
-
*.parquet filter=lfs diff=lfs merge=lfs -text
|
19 |
-
*.pb filter=lfs diff=lfs merge=lfs -text
|
20 |
-
*.pickle filter=lfs diff=lfs merge=lfs -text
|
21 |
-
*.pkl filter=lfs diff=lfs merge=lfs -text
|
22 |
-
*.pt filter=lfs diff=lfs merge=lfs -text
|
23 |
-
*.pth filter=lfs diff=lfs merge=lfs -text
|
24 |
-
*.rar filter=lfs diff=lfs merge=lfs -text
|
25 |
-
*.safetensors filter=lfs diff=lfs merge=lfs -text
|
26 |
-
saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
27 |
-
*.tar.* filter=lfs diff=lfs merge=lfs -text
|
28 |
-
*.tar filter=lfs diff=lfs merge=lfs -text
|
29 |
-
*.tflite filter=lfs diff=lfs merge=lfs -text
|
30 |
-
*.tgz filter=lfs diff=lfs merge=lfs -text
|
31 |
-
*.wasm filter=lfs diff=lfs merge=lfs -text
|
32 |
-
*.xz filter=lfs diff=lfs merge=lfs -text
|
33 |
-
*.zip filter=lfs diff=lfs merge=lfs -text
|
34 |
-
*.zst filter=lfs diff=lfs merge=lfs -text
|
35 |
-
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
.gitignore
CHANGED
@@ -2,16 +2,16 @@ run_*.sh
|
|
2 |
log/
|
3 |
saves
|
4 |
saves/
|
5 |
-
weights/
|
6 |
-
weights
|
7 |
output/
|
8 |
output
|
9 |
pretrained/
|
10 |
workspace
|
11 |
workspace/
|
12 |
-
ext_weights/
|
13 |
-
ext_weights
|
14 |
.checkpoints/
|
|
|
|
|
|
|
|
|
15 |
|
16 |
# Byte-compiled / optimized / DLL files
|
17 |
__pycache__/
|
|
|
2 |
log/
|
3 |
saves
|
4 |
saves/
|
|
|
|
|
5 |
output/
|
6 |
output
|
7 |
pretrained/
|
8 |
workspace
|
9 |
workspace/
|
|
|
|
|
10 |
.checkpoints/
|
11 |
+
weights/
|
12 |
+
ext_weights/
|
13 |
+
*.pth
|
14 |
+
*.pt
|
15 |
|
16 |
# Byte-compiled / optimized / DLL files
|
17 |
__pycache__/
|
app.py
CHANGED
@@ -54,6 +54,9 @@ def get_model() -> tuple[MMAudio, FeaturesUtils, SequenceConfig]:
|
|
54 |
|
55 |
|
56 |
net, feature_utils, seq_cfg = get_model()
|
|
|
|
|
|
|
57 |
|
58 |
|
59 |
@spaces.GPU
|
|
|
54 |
|
55 |
|
56 |
net, feature_utils, seq_cfg = get_model()
|
57 |
+
net.preprocess_conditions = torch.compile(net.preprocess_conditions)
|
58 |
+
net.predict_flow = torch.compile(net.predict_flow)
|
59 |
+
feature_utils.compile()
|
60 |
|
61 |
|
62 |
@spaces.GPU
|
requirements.txt
CHANGED
@@ -1,4 +1,4 @@
|
|
1 |
-
torch
|
2 |
torchaudio
|
3 |
torchvision
|
4 |
python-dotenv
|
|
|
1 |
+
torch >= 2.5.1
|
2 |
torchaudio
|
3 |
torchvision
|
4 |
python-dotenv
|