spandrel_extra_arches.architectures.CodeFormer.CodeFormer

class documentation

class CodeFormer(VQAutoEncoder): (source)

Constructor: CodeFormer(dim_embd, n_head, n_layers, codebook_size, ...)

View In Hierarchy

Undocumented

Method __init__ Undocumented
Method forward Undocumented
Class Variable hyperparameters Undocumented
Instance Variable channels Undocumented
Instance Variable connect_list Undocumented
Instance Variable dim_embd Undocumented
Instance Variable dim_mlp Undocumented
Instance Variable feat_emb Undocumented
Instance Variable ft_layers Undocumented
Instance Variable fuse_convs_dict Undocumented
Instance Variable fuse_encoder_block Undocumented
Instance Variable fuse_generator_block Undocumented
Instance Variable idx_pred_layer Undocumented
Instance Variable n_layers Undocumented
Instance Variable position_emb Undocumented
Method _init_weights Undocumented
def __init__(self, *, dim_embd=512, n_head=8, n_layers=9, codebook_size=1024, latent_size=256, connect_list=['32', '64', '128', '256'], fix_modules=['quantize', 'generator']): (source)
overrides spandrel_extra_arches.architectures.CodeFormer.__arch.codeformer.VQAutoEncoder.__init__

Undocumented

def forward(self, x, weight=0.5, **kwargs): (source)
overrides spandrel_extra_arches.architectures.CodeFormer.__arch.codeformer.VQAutoEncoder.forward

Undocumented

hyperparameters: dict = (source)

Undocumented

channels: dict[str, int] = (source)

Undocumented

connect_list = (source)

Undocumented

dim_embd = (source)

Undocumented

dim_mlp = (source)

Undocumented

feat_emb = (source)

Undocumented

ft_layers = (source)

Undocumented

fuse_convs_dict = (source)

Undocumented

fuse_encoder_block: dict[str, int] = (source)

Undocumented

fuse_generator_block: dict[str, int] = (source)

Undocumented

idx_pred_layer = (source)

Undocumented

n_layers = (source)

Undocumented

position_emb = (source)

Undocumented

def _init_weights(self, module): (source)

Undocumented