File size: 721 Bytes
99bbd30
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
## Backbones quick introduction


### unett.py
- flat unet transformer
- structure same as in e2-tts & voicebox paper except using rotary pos emb
- update: allow possible abs pos emb & convnextv2 blocks for embedded text before concat

### dit.py
- adaln-zero dit
- embedded timestep as condition
- concatted noised_input + masked_cond + embedded_text, linear proj in

- possible abs pos emb & convnextv2 blocks for embedded text before concat

- possible long skip connection (first layer to last layer)



### mmdit.py

- sd3 structure

- timestep as condition

- left stream: text embedded and applied a abs pos emb

- right stream: masked_cond & noised_input concatted and with same conv pos emb as unett