Functions — github.com/TencentARC/AnimeGamer

Method__init__

(self, in_channels: int, alpha: float = 0, merge_strategy: str = "learned")

VDM_Decoder/sgm/modules/autoencoding/temporal_ae.py:109

Method__init__

(self, in_channels: int, alpha: float = 0, merge_strategy: str = "learned")

VDM_Decoder/sgm/modules/autoencoding/temporal_ae.py:175

Method__init__

( self, *args, video_kernel_size: Union[int, list] = 3, alpha: float = 0.0,

VDM_Decoder/sgm/modules/autoencoding/temporal_ae.py:282

Method__init__

( self, *, dim=None, codebook_size=None, entropy_loss_weight=0.1,

VDM_Decoder/sgm/modules/autoencoding/regularizers/lookup_free_quantization.py:63

Method__init__

(self)

VDM_Decoder/sgm/modules/autoencoding/regularizers/base.py:10

Method__init__

(self, sample: bool = True)

VDM_Decoder/sgm/modules/autoencoding/regularizers/__init__.py:13

Method__init__

( self, levels: List[int], dim: Optional[int] = None, num_codebooks=1,

VDM_Decoder/sgm/modules/autoencoding/regularizers/finite_scalar_quantization.py:51

Method__init__

(self)

VDM_Decoder/sgm/modules/autoencoding/regularizers/quantize.py:18

Method__init__

( self, num_hiddens: int, embedding_dim: int, n_embed: int, straight_t

VDM_Decoder/sgm/modules/autoencoding/regularizers/quantize.py:69

Method__init__

( self, n_e: int, e_dim: int, beta: float = 0.25, remap: Optional[str]

VDM_Decoder/sgm/modules/autoencoding/regularizers/quantize.py:175

Method__init__

( self, n_embed: int, embedding_dim: int, beta: float, decay: float =

VDM_Decoder/sgm/modules/autoencoding/regularizers/quantize.py:326

Method__init__

( self, input_dim: int, n_codes: int, codebook_dim: int, beta: float =

VDM_Decoder/sgm/modules/autoencoding/regularizers/quantize.py:417

Method__init__

( self, f_channels, zq_channels, norm_layer=nn.GroupNorm, freeze_norm_

VDM_Decoder/sgm/modules/autoencoding/vqvae/movq_modules.py:35

Method__init__

(self, in_channels, with_conv)

VDM_Decoder/sgm/modules/autoencoding/vqvae/movq_modules.py:93

Method__init__

( self, *, in_channels, out_channels=None, conv_shortcut=False,

VDM_Decoder/sgm/modules/autoencoding/vqvae/movq_modules.py:111

Method__init__

(self, in_channels, zq_ch=None, add_conv=False)

VDM_Decoder/sgm/modules/autoencoding/vqvae/movq_modules.py:165

Method__init__

( self, *, ch, out_ch, ch_mult=(1, 2, 4, 8), num_res_blocks,

VDM_Decoder/sgm/modules/autoencoding/vqvae/movq_modules.py:203

Method__init__

( self, f_channels, zq_channels, norm_layer=nn.GroupNorm, freeze_norm_

VDM_Decoder/sgm/modules/autoencoding/vqvae/movq_dec_3d.py:49

Method__init__

( self, *, in_channels, out_channels=None, conv_shortcut=False,

VDM_Decoder/sgm/modules/autoencoding/vqvae/movq_dec_3d.py:101

Method__init__

( self, *, ch, out_ch, ch_mult=(1, 2, 4, 8), num_res_blocks,

VDM_Decoder/sgm/modules/autoencoding/vqvae/movq_dec_3d.py:200

Method__init__

( self, *, ch, out_ch, ch_mult=(1, 2, 4, 8), num_res_blocks,

VDM_Decoder/sgm/modules/autoencoding/vqvae/movq_dec_3d.py:338

Method__init__

(self, in_channels, with_conv)

VDM_Decoder/sgm/modules/autoencoding/vqvae/vqvae_blocks.py:39

Method__init__

(self, in_channels, with_conv)

VDM_Decoder/sgm/modules/autoencoding/vqvae/vqvae_blocks.py:53

Method__init__

(self, *, in_channels, out_channels=None, conv_shortcut=False, dropout, temb_channels=512)

VDM_Decoder/sgm/modules/autoencoding/vqvae/vqvae_blocks.py:71

Method__init__

(self, in_channels)

VDM_Decoder/sgm/modules/autoencoding/vqvae/vqvae_blocks.py:115

Method__init__

( self, *, ch, out_ch, ch_mult=(1, 2, 4, 8), num_res_blocks,

VDM_Decoder/sgm/modules/autoencoding/vqvae/vqvae_blocks.py:299

Method__init__

(self, n_e, e_dim, beta, remap=None, unknown_index="random", sane_index_shape=False, legacy=True)

VDM_Decoder/sgm/modules/autoencoding/vqvae/quantize.py:18

Method__init__

( self, chan_in, chan_out, kernel_size: Union[int, Tuple[int, int, int]], pad_mode="constant", **kwarg

VDM_Decoder/sgm/modules/autoencoding/vqvae/movq_enc_3d.py:53

Method__init__

(self, in_channels, with_conv, compress_time=False)

VDM_Decoder/sgm/modules/autoencoding/vqvae/movq_enc_3d.py:109

Method__init__

(self, in_channels, with_conv, compress_time=False, out_channels=None)

VDM_Decoder/sgm/modules/autoencoding/vqvae/movq_enc_3d.py:145

Method__init__

( self, *, in_channels, out_channels=None, conv_shortcut=False, dropout, temb_channels=512, pad_mode="

VDM_Decoder/sgm/modules/autoencoding/vqvae/movq_enc_3d.py:184

Method__init__

(self, in_channels)

VDM_Decoder/sgm/modules/autoencoding/vqvae/movq_enc_3d.py:246

Method__init__

(self, num_features, logdet=False, affine=True, allow_reverse_init=False)

VDM_Decoder/sgm/modules/autoencoding/lpips/util.py:46

Method__init__

Construct a PatchGAN discriminator Parameters: input_nc (int) -- the number of channels in input images ndf (int)

VDM_Decoder/sgm/modules/autoencoding/lpips/model/model.py:25

Method__init__

(self)

VDM_Decoder/sgm/modules/autoencoding/lpips/loss/lpips.py:59

Method__init__

(self, chn_in, chn_out=1, use_dropout=False)

VDM_Decoder/sgm/modules/autoencoding/lpips/loss/lpips.py:71

Method__init__

(self, requires_grad=False, pretrained=True)

VDM_Decoder/sgm/modules/autoencoding/lpips/loss/lpips.py:87

Method__init__

(self, input_channels, filters, downsample=True, antialiased_downsample=True)

VDM_Decoder/sgm/modules/autoencoding/losses/video_loss.py:114

Method__init__

( self, *, dim, image_size, channels=3, max_dim=512, a

VDM_Decoder/sgm/modules/autoencoding/losses/video_loss.py:152

Method__init__

( self, input_channels, filters, antialiased_downsample=True, )

VDM_Decoder/sgm/modules/autoencoding/losses/video_loss.py:226

Method__init__

( self, input_channels, filters, antialiased_downsample=True, pad_mode

VDM_Decoder/sgm/modules/autoencoding/losses/video_loss.py:265

Method__init__

( self, *, dim, image_size, frame_num, channels=3, max

VDM_Decoder/sgm/modules/autoencoding/losses/video_loss.py:313

Method__init__

( self, *, dim, image_size, frame_num, channels=3, max

VDM_Decoder/sgm/modules/autoencoding/losses/video_loss.py:408

Method__init__

( self, disc_start, perceptual_weight=1, adversarial_loss_weight=0, mu

VDM_Decoder/sgm/modules/autoencoding/losses/video_loss.py:503

Method__init__

( self, disc_start: int, logvar_init: float = 0.0, disc_num_layers: int = 3,

VDM_Decoder/sgm/modules/autoencoding/losses/discriminator_loss.py:18

Method__init__

( self, decoder_config, perceptual_weight=1.0, latent_weight=1.0, scal

VDM_Decoder/sgm/modules/autoencoding/losses/lpips.py:9

Method__init__

(self, value)

VDM_Decoder/sgm/modules/distributions/distributions.py:14

Method__init__

(self, parameters, deterministic=False)

VDM_Decoder/sgm/modules/distributions/distributions.py:25

Method__init__

(self)

VDM_Decoder/sgm/modules/encoders/ip_encoder.py:10

Method__init__

(self, arch="ViT-B-16", version="laion2b_s34b_b88k", device="cuda", freeze=True, layer="poole

VDM_Decoder/sgm/modules/encoders/ip_encoder.py:57

Method__init__

(self, emb_models: Union[List, ListConfig], cor_embs=[], cor_p=[])

VDM_Decoder/sgm/modules/encoders/modules.py:77

Method__init__

( self, model_dir="google/t5-v1_1-xxl", device="cuda", max_length=77,

VDM_Decoder/sgm/modules/encoders/modules.py:237

Method__init__

(self, args, transformer=None, params_dtype=torch.float, **kwargs)

VDM_Decoder/sgm/models/sat_ours/base_model.py:71

Method__init__

(self, hidden_size, num_attention_heads, attention_dropout_prob, output_dropout_prob,

VDM_Decoder/sgm/models/sat_ours/transformer.py:35

Method__init__

(self, hidden_size, num_attention_heads, attention_dropout_prob, output_dropout_prob, init_method,

VDM_Decoder/sgm/models/sat_ours/transformer.py:128

Method__init__

( self, hidden_size, num_attention_heads, attention_dropout_pr

VDM_Decoder/sgm/models/sat_ours/transformer.py:320

Method__init__

(self, num_layers, vocab_size, hidden_size,

VDM_Decoder/sgm/models/sat_ours/transformer.py:442

Method__init__

(self, chan_in, chan_out, kernel_size: Union[int, Tuple[int, int, int]], stride=1, **kwargs)

VDM_Decoder/vae_modules/cp_enc_dec.py:361

Method__init__

( self, f_channels, zq_channels, freeze_norm_layer=False, add_conv=Fal

VDM_Decoder/vae_modules/cp_enc_dec.py:452

Method__init__

(self, in_channels, with_conv, compress_time=False, out_channels=None)

VDM_Decoder/vae_modules/cp_enc_dec.py:572

Method__init__

( self, *, in_channels, out_channels=None, conv_shortcut=False,

VDM_Decoder/vae_modules/cp_enc_dec.py:615

Method__init__

( self, *, ch, out_ch, ch_mult=(1, 2, 4, 8), num_res_blocks,

VDM_Decoder/vae_modules/cp_enc_dec.py:717

Method__init__

( self, *, ch, out_ch, ch_mult=(1, 2, 4, 8), num_res_blocks,

VDM_Decoder/vae_modules/cp_enc_dec.py:840

Method__init__

( self, ema_decay: Union[None, float] = None, monitor: Union[None, str] = None,

VDM_Decoder/vae_modules/autoencoder.py:39

Method__init__

( self, *args, encoder_config: Dict, decoder_config: Dict, loss_config

VDM_Decoder/vae_modules/autoencoder.py:135

Method__init__

(self, embed_dim: int, **kwargs)

VDM_Decoder/vae_modules/autoencoder.py:434

Method__init__

(self, *args, **kwargs)

VDM_Decoder/vae_modules/autoencoder.py:513

Method__init__

( self, ckpt_path: Union[None, str] = None, ignore_keys: Union[Tuple, list] = (),

VDM_Decoder/vae_modules/autoencoder.py:527

Method__init__

( self, cp_size=0, *args, **kwargs, )

VDM_Decoder/vae_modules/autoencoder.py:579

Method__init__

(self, parameters, deterministic=False)

VDM_Decoder/vae_modules/regularizers.py:11

Method__init__

(self, sample: bool = True)

VDM_Decoder/vae_modules/regularizers.py:91

Method__init__

(self, tokenizer, num_img_gen_tokens=64)

MLLM/src/models_clm/generation.py:40

Method__init__

(self, llm, input_resampler, output_resampler, num_frames=4, query_pos=False, learnable_pos=False, use_kl

MLLM/src/models_clm/models.py:145

Method__init__

( self, dim, num_heads=8, qkv_bias=False, qk_scale=None, attn_

MLLM/src/models/vit.py:48

Method__init__

( self, dim, num_heads, mlp_ratio=4.0, qkv_bias=False, qk_scal

MLLM/src/models/vit.py:124

Method__init__

(self, num_queries, embed_dim, input_dim, output_dim)

MLLM/src/models/discrete_models.py:15

Method__init__

( self, vocab_size=32000, # Vocabulary size, default is 32000. hidden_size=4096, # S

MLLM/src/models/mistral/configuration_mistral.py:22

Method__init__

(self, is_causal: bool, sliding_window: Optional[int] = None)

MLLM/src/models/mistral/modeling_attn_mask_utils.py:33

Method__init__

MistralRMSNorm is equivalent to T5LayerNorm

MLLM/src/models/mistral/modeling_mistral.py:82

Method__init__

(self, dim, max_position_embeddings=2048, base=10000, device=None)

MLLM/src/models/mistral/modeling_mistral.py:101

Method__init__

(self, config: MistralConfig, layer_idx: Optional[int] = None)

MLLM/src/models/mistral/modeling_mistral.py:208

Method__init__

(self, *args, **kwargs)

MLLM/src/models/mistral/modeling_mistral.py:335

Method__init__

(self, config: MistralConfig, layer_idx: int)

MLLM/src/models/mistral/modeling_mistral.py:686

Method__init__

(self, config: MistralConfig)

MLLM/src/models/mistral/modeling_mistral.py:863

Method__init__

(self, config)

MLLM/src/models/mistral/modeling_mistral.py:1045

Function_find_children

Find all modules of a certain class (or union of classes). Returns all matching modules, along with the parent of those moduless and the

VDM_Decoder/sgm/modules/diffusionmodules/lora.py:217

Function_find_modules_v2

Find all modules of a certain class (or union of classes) that are direct or indirect descendants of other modules of a certain class (or uni

VDM_Decoder/sgm/modules/diffusionmodules/lora.py:234

Method_forward

(self, x, context=None, additional_tokens=None, n_times_crossframe_attn_in_self=0)

VDM_Decoder/sgm/modules/attention.py:682

Method_forward

(self, x, context=None)

VDM_Decoder/sgm/modules/attention.py:733

Method_forward

(self, x, emb)

VDM_Decoder/sgm/modules/diffusionmodules/openaimodel.py:318

Method_forward

(self, x)

VDM_Decoder/sgm/modules/diffusionmodules/openaimodel.py:390

Method_fuse_lora

(self, lora_scale=1.0)

VDM_Decoder/sgm/modules/diffusionmodules/lora.py:96

Method_fuse_lora

(self, lora_scale=1.0)

VDM_Decoder/sgm/modules/diffusionmodules/lora.py:166

Function_gather

(input_, dim)

VDM_Decoder/vae_modules/cp_enc_dec.py:102

Method_init_weights

(self, module)

MLLM/src/models/mistral/modeling_mistral.py:769

Method_make_attn

(self)

VDM_Decoder/sgm/modules/autoencoding/temporal_ae.py:306

Method_make_conv

(self)

VDM_Decoder/sgm/modules/autoencoding/temporal_ae.py:316

Method_make_resblock

(self)

VDM_Decoder/sgm/modules/autoencoding/temporal_ae.py:322

Method_reorder_cache

(past_key_values, beam_idx)

MLLM/src/models/mistral/modeling_mistral.py:1220

Method_shape

(self, tensor: torch.Tensor, seq_len: int, bsz: int)

MLLM/src/models/mistral/modeling_mistral.py:245

Functions1,003 in github.com/TencentARC/AnimeGamer