Method__init__(
self,
n_ctx: int,
width: int,
layers: int,
heads: int,
)
libcom/os_insert/source/ldm/modules/encoders/xf.py:105
Method__init__(self, n_embed, n_layer, vocab_size, max_seq_len=77, device="cuda")
libcom/os_insert/source/ldm/modules/encoders/modules.py:39
Method__init__(self, disc_start, codebook_weight=1.0, pixelloss_weight=1.0,
disc_num_layers=3, disc_in_chan
libcom/os_insert/source/ldm/modules/losses/vqperceptual.py:44
Method__init__(self, disc_start, logvar_init=0.0, kl_weight=1.0, pixelloss_weight=1.0,
disc_num_layers=3, d
libcom/os_insert/source/ldm/modules/losses/contperceptual.py:8
Method__init__(self, batch_size, num_workers=None, train=None, validation=None, test=None, predict=None,
wr
libcom/os_insert/source/ldm/data/datamodule.py:53