Add files using upload-large-folder tool
Browse files- README.md +110 -0
- checkpoints/TextConditionerA/config.yaml +22 -0
- checkpoints/TextConditionerA/weights.safetensors +3 -0
- checkpoints/TextConditionerV/config.yaml +22 -0
- checkpoints/TextConditionerV/weights.safetensors +3 -0
- checkpoints/Woosh-AE/config.yaml +27 -0
- checkpoints/Woosh-AE/weights.safetensors +3 -0
- checkpoints/Woosh-CLAP/config.yaml +25 -0
- checkpoints/Woosh-CLAP/weights_audio.safetensors +3 -0
- checkpoints/Woosh-CLAP/weights_text.safetensors +3 -0
- checkpoints/Woosh-DFlow/config.yaml +51 -0
- checkpoints/Woosh-DFlow/weights.safetensors +3 -0
- checkpoints/Woosh-DVFlow-8s/config.yaml +64 -0
- checkpoints/Woosh-DVFlow-8s/weights.safetensors +3 -0
- checkpoints/Woosh-Flow/config.yaml +48 -0
- checkpoints/Woosh-Flow/weights.safetensors +3 -0
- checkpoints/Woosh-VFlow-8s/config.yaml +60 -0
- checkpoints/Woosh-VFlow-8s/weights.safetensors +3 -0
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
|