AEmotionStudio commited on
Commit
f7b524d
·
verified ·
1 Parent(s): affb8ed

Add files using upload-large-folder tool

Browse files
README.md ADDED
@@ -0,0 +1,110 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: cc-by-nc-4.0
3
+ library_name: woosh
4
+ tags:
5
+ - audio
6
+ - audio-generation
7
+ - sound-effects
8
+ - foley
9
+ - text-to-audio
10
+ - video-to-audio
11
+ - t2a
12
+ - v2a
13
+ - safetensors
14
+ - woosh
15
+ pipeline_tag: text-to-audio
16
+ ---
17
+
18
+ # Woosh — Sony AI Sound-Effect Foundation Model (Mirror)
19
+
20
+ This repository is a **community mirror** of the open weights released by Sony Research for [Woosh](https://github.com/SonyResearch/Woosh) — a foundation model for sound-effect generation supporting text-to-audio (T2A) and video-to-audio (V2A) synthesis.
21
+
22
+ All files here are a one-to-one copy of Sony's [v1.0.0 GitHub release](https://github.com/SonyResearch/Woosh/releases/tag/v1.0.0), repackaged into a single browseable HF repo for convenience.
23
+
24
+ ## License — CC-BY-NC 4.0 (Non-Commercial)
25
+
26
+ > All open weights in this repository are released by Sony Research under the [CC-BY-NC 4.0](https://creativecommons.org/licenses/by-nc/4.0/) license. **Generated outputs inherit the non-commercial restriction.** You may not use model outputs in commercial products, paid releases, or client work. The upstream project's source code is released separately under MIT / Apache-2.0.
27
+
28
+ If you need to attribute: **Sony Research — Woosh** ([arXiv / paper](https://github.com/SonyResearch/Woosh), GitHub: `SonyResearch/Woosh`).
29
+
30
+ ## Model Suite
31
+
32
+ Woosh is a multi-model suite. All components are required together — the generative backbones depend on the shared AE, CLAP, and text conditioners at inference time.
33
+
34
+ ### Shared infrastructure
35
+
36
+ | Folder | Role | File(s) |
37
+ |---|---|---|
38
+ | `checkpoints/Woosh-AE/` | Audio encoder / decoder producing high-quality latents | `weights.safetensors`, `config.yaml` |
39
+ | `checkpoints/Woosh-CLAP/` | Multimodal text-audio alignment model (audio + text encoders) | `weights_audio.safetensors`, `weights_text.safetensors`, `config.yaml` |
40
+ | `checkpoints/TextConditionerA/` | Text conditioner for the T2A path (pairs with Flow / DFlow) | `weights.safetensors`, `config.yaml` |
41
+ | `checkpoints/TextConditionerV/` | Text conditioner for the V2A path (pairs with VFlow / DVFlow) | `weights.safetensors`, `config.yaml` |
42
+
43
+ ### Generative backbones
44
+
45
+ | Folder | Task | Notes |
46
+ |---|---|---|
47
+ | `checkpoints/Woosh-Flow/` | Text → Audio | Full-quality T2A latent diffusion |
48
+ | `checkpoints/Woosh-DFlow/` | Text → Audio | Distilled T2A — fewer steps, faster inference |
49
+ | `checkpoints/Woosh-VFlow-8s/` | Video → Audio | V2A latent diffusion — **fixed 8-second output** |
50
+ | `checkpoints/Woosh-DVFlow-8s/` | Video → Audio | Distilled V2A — fewer steps, **fixed 8-second output** |
51
+
52
+ Every weight file ships as `safetensors`. No `.pt` / `.ckpt` / `.bin` in this mirror.
53
+
54
+ ## Layout
55
+
56
+ ```
57
+ checkpoints/
58
+ ├── Woosh-AE/
59
+ │ ├── weights.safetensors
60
+ │ └── config.yaml
61
+ ├── Woosh-CLAP/
62
+ │ ├── weights_audio.safetensors
63
+ │ ├── weights_text.safetensors
64
+ │ └── config.yaml
65
+ ├── TextConditionerA/
66
+ │ ├── weights.safetensors
67
+ │ └── config.yaml
68
+ ├── TextConditionerV/
69
+ │ ├── weights.safetensors
70
+ │ └── config.yaml
71
+ ├── Woosh-Flow/
72
+ │ ├── weights.safetensors
73
+ │ └── config.yaml
74
+ ├── Woosh-DFlow/
75
+ │ ├── weights.safetensors
76
+ │ └── config.yaml
77
+ ├── Woosh-VFlow-8s/
78
+ │ ├── weights.safetensors
79
+ │ └── config.yaml
80
+ └── Woosh-DVFlow-8s/
81
+ ├── weights.safetensors
82
+ └── config.yaml
83
+ ```
84
+
85
+ Directory names match Sony's release zip layout exactly so the upstream inference code finds its configs without modification.
86
+
87
+ ## Usage
88
+
89
+ This mirror is intended to be consumed by [Sony's upstream `woosh` package](https://github.com/SonyResearch/Woosh). Clone and install the upstream repo, then point it at a local copy of this mirror's `checkpoints/` directory.
90
+
91
+ ```bash
92
+ # Clone upstream
93
+ git clone https://github.com/SonyResearch/Woosh.git
94
+ cd Woosh
95
+
96
+ # Sony's suggested env setup (uses uv)
97
+ uv sync
98
+ uv pip install -e .
99
+
100
+ # Pull weights from this mirror
101
+ hf download AEmotionStudio/woosh-models --local-dir ./
102
+ ```
103
+
104
+ ## Acknowledgements
105
+
106
+ All credit for the Woosh models belongs to **Sony Research**. This mirror exists solely to make the CC-BY-NC open weights easier to fetch and integrate. Please cite the upstream project and respect the non-commercial license.
107
+
108
+ - Upstream: <https://github.com/SonyResearch/Woosh>
109
+ - Release: <https://github.com/SonyResearch/Woosh/releases/tag/v1.0.0>
110
+ - License: [CC-BY-NC 4.0](https://creativecommons.org/licenses/by-nc/4.0/)
checkpoints/TextConditionerA/config.yaml ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ exclude_from_checkpoint: false
2
+ trainable: false
3
+ sentence_config:
4
+ frozen: false
5
+ model: roberta-large
6
+ max_sentence_tokens: 77
7
+ adopt_n_layers: 0
8
+ adopt_layer_size: 2048
9
+ pool_type: eos
10
+ add_pooling_layer: true
11
+ hidden_dropout_prob: 0.1
12
+ attention_probs_dropout_prob: 0.1
13
+ finetune_n_layers: -1
14
+ last_hidden_state: true
15
+ use_shared_space: false
16
+ normalize_shared_space: true
17
+ freeze_clap: true
18
+ lhs_index: -2
19
+ remove_special_tokens: false
20
+ eval_mode: true
21
+ text_preprocessing: no_op
22
+ shared_representation_size: 1024
checkpoints/TextConditionerA/weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9f66e1877efd49e3198068ffac74d00a84df18e53189643373e2e777224856bb
3
+ size 1425689504
checkpoints/TextConditionerV/config.yaml ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ exclude_from_checkpoint: false
2
+ trainable: false
3
+ sentence_config:
4
+ frozen: false
5
+ model: roberta-large
6
+ max_sentence_tokens: 77
7
+ adopt_n_layers: 0
8
+ adopt_layer_size: 2048
9
+ pool_type: eos
10
+ add_pooling_layer: true
11
+ hidden_dropout_prob: 0.1
12
+ attention_probs_dropout_prob: 0.1
13
+ finetune_n_layers: -1
14
+ last_hidden_state: true
15
+ use_shared_space: false
16
+ normalize_shared_space: true
17
+ freeze_clap: true
18
+ lhs_index: -2
19
+ remove_special_tokens: false
20
+ eval_mode: true
21
+ text_preprocessing: no_op
22
+ shared_representation_size: 1024
checkpoints/TextConditionerV/weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b0dd912c322582cb95531de69cf1472c90752e468087a3b0fc461dea49975e80
3
+ size 1425689504
checkpoints/Woosh-AE/config.yaml ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ exclude_from_checkpoint: true
2
+ trainable: true
3
+ istft_head:
4
+ _target_: woosh.module.model.vocos.ISTFTCircleHead
5
+ _partial_: true
6
+ padding: center
7
+ conv_pad: same
8
+ conv_kernel: 1
9
+ softclip: softplus
10
+ ztransform:
11
+ _target_: woosh.module.model.vocos.ZeroDropoutTransform
12
+ _partial_: true
13
+ p: 1.0
14
+ _target_: woosh.module.model.VocosAutoEncoder
15
+ channels: 1
16
+ z_dim: 128
17
+ d_model: 2048
18
+ intermediate_dim: 3072
19
+ n_fft: 960
20
+ hop_length: 480
21
+ num_layers: 8
22
+ enc_num_layers: 8
23
+ input_layer_norm: true
24
+ final_layer_norm: true
25
+ stft_normalized: false
26
+ spec_embed: stft-complex
27
+ sample_rate: 48000
checkpoints/Woosh-AE/weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:48e0d864beee90b17d232c74b16e9878dadc73b4ae55e4f6939df3f559c5d88a
3
+ size 884664420
checkpoints/Woosh-CLAP/config.yaml ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ shared_representation_size: 1024
2
+ text_preprocessing: no_op
3
+ audio:
4
+ name: passt_no
5
+ sample_rate: 32000
6
+ segment_length: 5
7
+ embedding_dim: 128
8
+ aggregate: mean
9
+ type: embedding
10
+ adopt_n_layers: 0
11
+ adopt_layer_size: 2048
12
+ s_patchout_t: 0
13
+ s_patchout_f: 0
14
+ eval_max_sec: 60
15
+ mask_attention: false
16
+ sentence:
17
+ model: roberta-large
18
+ max_sentence_tokens: 77
19
+ adopt_n_layers: 0
20
+ adopt_layer_size: 2048
21
+ pool_type: eos
22
+ add_pooling_layer: true
23
+ hidden_dropout_prob: 0.1
24
+ attention_probs_dropout_prob: 0.1
25
+ normalize: true
checkpoints/Woosh-CLAP/weights_audio.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d32f248f77f34cac2b6b71f13d4c1fb01207b1c83fc58dd130895c894f588d85
3
+ size 347657136
checkpoints/Woosh-CLAP/weights_text.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9f66e1877efd49e3198068ffac74d00a84df18e53189643373e2e777224856bb
3
+ size 1425689504
checkpoints/Woosh-DFlow/config.yaml ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ exclude_from_checkpoint: false
2
+ trainable: true
3
+ ldm:
4
+ exclude_from_checkpoint: false
5
+ trainable: true
6
+ model_type: LatentDiffusionModel
7
+ dit:
8
+ model_type: mmmssflux
9
+ rope_len_multiplier: 2
10
+ max_description_length: 77
11
+ max_seq_len: 501
12
+ dim: 1024
13
+ inter_dim: 4096
14
+ fixed_timestep_features: false
15
+ timestep_features_dim: 256
16
+ n_layers: 12
17
+ n_heads: 8
18
+ n_multimodal_layers: 6
19
+ qk_nope_head_dim: 16
20
+ qk_rope_head_dim: 112
21
+ qkv_head_dim: 128
22
+ n_memory_tokens_rope: 1
23
+ n_memory_tokens_description: 0
24
+ original_seq_len: 501
25
+ rope_theta: 10000.0
26
+ rope_factor: 40.0
27
+ beta_fast: 32
28
+ beta_slow: 1
29
+ io_channels: 128
30
+ cond_token_dim: 1024
31
+ adaln_last_layer: true
32
+ adaln_last_layer_nomod: false
33
+ non_checkpoint_layers: 14
34
+ mask_out_before: -1
35
+ estimate_logvar: true
36
+ no_description_mask: true
37
+ symmetric_attention_init: false
38
+ patch_size: 1
39
+ num_sinks: 0
40
+ mlp_act: gelu
41
+ conditioners:
42
+ text:
43
+ exclude_from_checkpoint: true
44
+ trainable: false
45
+ path: checkpoints/TextConditionerA
46
+ autoencoder:
47
+ exclude_from_checkpoint: true
48
+ trainable: false
49
+ path: checkpoints/Woosh-AE
50
+ sigma_data: 1.0
51
+ pred_type: v_pred
checkpoints/Woosh-DFlow/weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c9fc4a2220fa4ba54e01611afe58e061854a58ad559eb56de2cca39112f7c8ef
3
+ size 1378890620
checkpoints/Woosh-DVFlow-8s/config.yaml ADDED
@@ -0,0 +1,64 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ exclude_from_checkpoint: false
2
+ trainable: true
3
+ pretrained_model_type: videokontext
4
+ ldm:
5
+ exclude_from_checkpoint: false
6
+ trainable: true
7
+ model_type: VideoKontextLDM
8
+ ldm:
9
+ exclude_from_checkpoint: false
10
+ trainable: true
11
+ dit:
12
+ model_type: mmmssflux
13
+ max_description_length: 77
14
+ max_seq_len: 501
15
+ rope_len_multiplier: 2
16
+ dim: 1024
17
+ inter_dim: 4096
18
+ fixed_timestep_features: false
19
+ timestep_features_dim: 256
20
+ n_layers: 12
21
+ n_heads: 8
22
+ n_multimodal_layers: 6
23
+ qk_nope_head_dim: 16
24
+ qk_rope_head_dim: 112
25
+ qkv_head_dim: 128
26
+ n_memory_tokens_rope: 1
27
+ n_memory_tokens_description: 0
28
+ original_seq_len: 501
29
+ rope_theta: 10000.0
30
+ rope_factor: 40.0
31
+ beta_fast: 32
32
+ beta_slow: 1
33
+ io_channels: 128
34
+ cond_token_dim: 1024
35
+ adaln_last_layer: true
36
+ adaln_last_layer_nomod: false
37
+ non_checkpoint_layers: 14
38
+ mask_out_before: -1
39
+ estimate_logvar: true
40
+ no_description_mask: true
41
+ symmetric_attention_init: false
42
+ patch_size: 1
43
+ num_sinks: 0
44
+ mlp_act: gelu
45
+ conditioners:
46
+ text:
47
+ exclude_from_checkpoint: true
48
+ trainable: false
49
+ path: checkpoints/TextConditionerV
50
+ autoencoder:
51
+ exclude_from_checkpoint: true
52
+ trainable: false
53
+ path: checkpoints/Woosh-AE
54
+ sigma_data: 1.0
55
+ pred_type: v_pred
56
+ audio_fps: 100
57
+ video_fps: 24
58
+ embed_dim: 768
59
+ embed_key: synch_out
60
+ pts_seconds_key: synch_pts_seconds
61
+ non_checkpoint_layers: 14
62
+ n_layers_encoder: 0
63
+ trainable_no_cond: false
64
+ use_batch_mask: true
checkpoints/Woosh-DVFlow-8s/weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c87b796798fd96a4a3b1deeb70ccd05c6dc782fe546b622ce1368a05e441cd30
3
+ size 1684261500
checkpoints/Woosh-Flow/config.yaml ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ exclude_from_checkpoint: false
2
+ trainable: true
3
+ model_type: LatentDiffusionModel
4
+ dit:
5
+ model_type: mmmssflux
6
+ rope_len_multiplier: 2
7
+ max_description_length: 77
8
+ max_seq_len: 501
9
+ dim: 1024
10
+ inter_dim: 4096
11
+ fixed_timestep_features: false
12
+ timestep_features_dim: 256
13
+ n_layers: 12
14
+ n_heads: 8
15
+ n_multimodal_layers: 6
16
+ qk_nope_head_dim: 16
17
+ qk_rope_head_dim: 112
18
+ qkv_head_dim: 128
19
+ n_memory_tokens_rope: 1
20
+ n_memory_tokens_description: 0
21
+ original_seq_len: 501
22
+ rope_theta: 10000.0
23
+ rope_factor: 40.0
24
+ beta_fast: 32
25
+ beta_slow: 1
26
+ io_channels: 128
27
+ cond_token_dim: 1024
28
+ adaln_last_layer: true
29
+ adaln_last_layer_nomod: false
30
+ non_checkpoint_layers: 14
31
+ mask_out_before: -1
32
+ estimate_logvar: true
33
+ no_description_mask: true
34
+ symmetric_attention_init: false
35
+ patch_size: 1
36
+ num_sinks: 0
37
+ mlp_act: gelu
38
+ conditioners:
39
+ text:
40
+ exclude_from_checkpoint: true
41
+ trainable: false
42
+ path: checkpoints/TextConditionerA
43
+ autoencoder:
44
+ exclude_from_checkpoint: true
45
+ trainable: false
46
+ path: checkpoints/Woosh-AE
47
+ sigma_data: 1.0
48
+ pred_type: v_pred
checkpoints/Woosh-Flow/weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4e2eafa2f9063934859888a2edd25ad4485f82dca4caaa7720f98d832f7ca39b
3
+ size 1348979404
checkpoints/Woosh-VFlow-8s/config.yaml ADDED
@@ -0,0 +1,60 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ exclude_from_checkpoint: false
2
+ trainable: true
3
+ model_type: VideoKontextLDM
4
+ ldm:
5
+ exclude_from_checkpoint: false
6
+ trainable: true
7
+ dit:
8
+ model_type: mmmssflux
9
+ max_description_length: 77
10
+ max_seq_len: 501
11
+ rope_len_multiplier: 2
12
+ dim: 1024
13
+ inter_dim: 4096
14
+ fixed_timestep_features: false
15
+ timestep_features_dim: 256
16
+ n_layers: 12
17
+ n_heads: 8
18
+ n_multimodal_layers: 6
19
+ qk_nope_head_dim: 16
20
+ qk_rope_head_dim: 112
21
+ qkv_head_dim: 128
22
+ n_memory_tokens_rope: 1
23
+ n_memory_tokens_description: 0
24
+ original_seq_len: 501
25
+ rope_theta: 10000.0
26
+ rope_factor: 40.0
27
+ beta_fast: 32
28
+ beta_slow: 1
29
+ io_channels: 128
30
+ cond_token_dim: 1024
31
+ adaln_last_layer: true
32
+ adaln_last_layer_nomod: false
33
+ non_checkpoint_layers: 14
34
+ mask_out_before: -1
35
+ estimate_logvar: true
36
+ no_description_mask: true
37
+ symmetric_attention_init: false
38
+ patch_size: 1
39
+ num_sinks: 0
40
+ mlp_act: gelu
41
+ conditioners:
42
+ text:
43
+ exclude_from_checkpoint: true
44
+ trainable: false
45
+ path: checkpoints/TextConditionerV
46
+ autoencoder:
47
+ exclude_from_checkpoint: true
48
+ trainable: false
49
+ path: checkpoints/Woosh-AE
50
+ sigma_data: 1.0
51
+ pred_type: v_pred
52
+ audio_fps: 100
53
+ video_fps: 24
54
+ embed_dim: 768
55
+ embed_key: synch_out
56
+ pts_seconds_key: synch_pts_seconds
57
+ non_checkpoint_layers: 14
58
+ n_layers_encoder: 0
59
+ trainable_no_cond: false
60
+ use_batch_mask: true
checkpoints/Woosh-VFlow-8s/weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ed2e7639e830060e532c955d4b4dc23869c8574605267846fa326c28730fcc7d
3
+ size 1654350276