Spaces:
Paused
Paused
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
|