Rex Cheng commited on
Commit
b7f72e1
1 Parent(s): 10fd932
Files changed (4) hide show
  1. .gitattributes +0 -35
  2. .gitignore +4 -4
  3. app.py +3 -0
  4. 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.4.0
2
  torchaudio
3
  torchvision
4
  python-dotenv
 
1
+ torch >= 2.5.1
2
  torchaudio
3
  torchvision
4
  python-dotenv