Method__init__(
self,
z_channels=256,
ch=128,
ch_mult=(1, 1, 2, 2, 4),
num_res_block
janus/models/vq_model.py:128
Method__init__(self, n_e, e_dim, beta, entropy_loss_ratio, l2_norm, show_usage)
janus/models/vq_model.py:218
Methodforward Args: images (torch.Tensor): [b, 3, H, W] Returns: image_features (torch.Tensor): [b, n_patch, d]
janus/janusflow/models/clip_encoder.py:107
Methodforward Args: images (torch.Tensor): [b, 3, H, W] Returns: image_features (torch.Tensor): [b, n_patch, d]
janus/models/clip_encoder.py:107