soundsol commited on
Commit
5b3d971
·
verified ·
1 Parent(s): 8772ff6

Upload config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.json +27 -0
config.json ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "audio-spatializer",
3
+ "architecture": "CrossAttnSpatializer",
4
+ "sample_rate": 24000,
5
+ "n_fft": 1024,
6
+ "hop_length": 512,
7
+ "n_mels": 128,
8
+ "audio_embed_dim": 512,
9
+ "text_embed_dim": 512,
10
+ "nhead": 8,
11
+ "num_layers": 6,
12
+ "output_channels": 4,
13
+ "output_format": "foa",
14
+ "training": {
15
+ "framework": "pytorch-lightning",
16
+ "final_epoch": 14,
17
+ "final_step": 342,
18
+ "checkpoint": "epoch=14-step=342.ckpt"
19
+ },
20
+ "spatial_parameters": {
21
+ "direction": ["front", "front-left", "left", "back-left", "back", "back-right", "right", "front-right"],
22
+ "elevation": ["down", "level", "up"],
23
+ "distance": ["near", "mid", "far"],
24
+ "room_size": ["small", "medium", "large"],
25
+ "reverb": ["dry", "medium", "wet"]
26
+ }
27
+ }