SYMBOL INDEX (133 symbols across 14 files)

FILE: animatediff/data/dataset.py
  class WebVid10M (line 13) | class WebVid10M(Dataset):
    method __init__ (line 14) | def __init__(
    method get_batch (line 39) | def get_batch(self, idx):
    method __len__ (line 63) | def __len__(self):
    method __getitem__ (line 66) | def __getitem__(self, idx):

FILE: animatediff/models/attention.py
  class Transformer3DModelOutput (line 20) | class Transformer3DModelOutput(BaseOutput):
  class Transformer3DModel (line 31) | class Transformer3DModel(ModelMixin, ConfigMixin):
    method __init__ (line 33) | def __init__(
    method forward (line 95) | def forward(self, hidden_states, encoder_hidden_states=None, timestep=...
  class BasicTransformerBlock (line 145) | class BasicTransformerBlock(nn.Module):
    method __init__ (line 146) | def __init__(
    method set_use_memory_efficient_attention_xformers (line 228) | def set_use_memory_efficient_attention_xformers(self, use_memory_effic...
    method forward (line 256) | def forward(self, hidden_states, encoder_hidden_states=None, timestep=...

FILE: animatediff/models/motion_module.py
  function zero_module (line 20) | def zero_module(module):
  class TemporalTransformer3DModelOutput (line 28) | class TemporalTransformer3DModelOutput(BaseOutput):
  function get_motion_module (line 39) | def get_motion_module(
  class VanillaTemporalModule (line 50) | class VanillaTemporalModule(nn.Module):
    method __init__ (line 51) | def __init__(
    method forward (line 79) | def forward(self, input_tensor, temb, encoder_hidden_states, attention...
  class TemporalTransformer3DModel (line 87) | class TemporalTransformer3DModel(nn.Module):
    method __init__ (line 88) | def __init__(
    method forward (line 136) | def forward(self, hidden_states, encoder_hidden_states=None, attention...
  class TemporalTransformerBlock (line 163) | class TemporalTransformerBlock(nn.Module):
    method __init__ (line 164) | def __init__(
    method forward (line 212) | def forward(self, hidden_states, encoder_hidden_states=None, attention...
  class PositionalEncoding (line 227) | class PositionalEncoding(nn.Module):
    method __init__ (line 228) | def __init__(
    method forward (line 243) | def forward(self, x):
  class VersatileAttention (line 248) | class VersatileAttention(CrossAttention):
    method __init__ (line 249) | def __init__(
    method extra_repr (line 269) | def extra_repr(self):
    method forward (line 272) | def forward(self, hidden_states, encoder_hidden_states=None, attention...

FILE: animatediff/models/resnet.py
  class InflatedConv3d (line 10) | class InflatedConv3d(nn.Conv2d):
    method forward (line 11) | def forward(self, x):
  class InflatedGroupNorm (line 21) | class InflatedGroupNorm(nn.GroupNorm):
    method forward (line 22) | def forward(self, x):
  class Upsample3D (line 32) | class Upsample3D(nn.Module):
    method __init__ (line 33) | def __init__(self, channels, use_conv=False, use_conv_transpose=False,...
    method forward (line 47) | def forward(self, hidden_states, output_size=None):
  class Downsample3D (line 83) | class Downsample3D(nn.Module):
    method __init__ (line 84) | def __init__(self, channels, use_conv=False, out_channels=None, paddin...
    method forward (line 98) | def forward(self, hidden_states):
  class ResnetBlock3D (line 109) | class ResnetBlock3D(nn.Module):
    method __init__ (line 110) | def __init__(
    method forward (line 182) | def forward(self, input_tensor, temb):
  class Mish (line 215) | class Mish(torch.nn.Module):
    method forward (line 216) | def forward(self, hidden_states):

FILE: animatediff/models/sparse_controlnet.py
  class SparseControlNetOutput (line 44) | class SparseControlNetOutput(BaseOutput):
  class SparseControlNetConditioningEmbedding (line 49) | class SparseControlNetConditioningEmbedding(nn.Module):
    method __init__ (line 50) | def __init__(
    method forward (line 72) | def forward(self, conditioning):
  class SparseControlNetModel (line 85) | class SparseControlNetModel(ModelMixin, ConfigMixin):
    method __init__ (line 89) | def __init__(
    method from_unet (line 317) | def from_unet(
    method image_layer_filter (line 373) | def image_layer_filter(state_dict):
    method set_attention_slice (line 381) | def set_attention_slice(self, slice_size):
    method _set_gradient_checkpointing (line 446) | def _set_gradient_checkpointing(self, module, value=False):
    method forward (line 450) | def forward(
  function zero_module (line 584) | def zero_module(module):

FILE: animatediff/models/unet.py
  class UNet3DConditionOutput (line 34) | class UNet3DConditionOutput(BaseOutput):
  class UNet3DConditionModel (line 38) | class UNet3DConditionModel(ModelMixin, ConfigMixin):
    method __init__ (line 42) | def __init__(
    method set_attention_slice (line 251) | def set_attention_slice(self, slice_size):
    method _set_gradient_checkpointing (line 316) | def _set_gradient_checkpointing(self, module, value=False):
    method forward (line 320) | def forward(
    method from_pretrained_2d (line 478) | def from_pretrained_2d(cls, pretrained_model_name_or_path, unet_additi...

FILE: animatediff/models/unet_blocks.py
  function get_down_block (line 12) | def get_down_block(
  function get_up_block (line 92) | def get_up_block(
  class UNetMidBlock3DCrossAttn (line 171) | class UNetMidBlock3DCrossAttn(nn.Module):
    method __init__ (line 172) | def __init__(
    method forward (line 271) | def forward(self, hidden_states, temb=None, encoder_hidden_states=None...
  class CrossAttnDownBlock3D (line 281) | class CrossAttnDownBlock3D(nn.Module):
    method __init__ (line 282) | def __init__(
    method forward (line 382) | def forward(self, hidden_states, temb=None, encoder_hidden_states=None...
  class DownBlock3D (line 424) | class DownBlock3D(nn.Module):
    method __init__ (line 425) | def __init__(
    method forward (line 493) | def forward(self, hidden_states, temb=None, encoder_hidden_states=None):
  class CrossAttnUpBlock3D (line 524) | class CrossAttnUpBlock3D(nn.Module):
    method __init__ (line 525) | def __init__(
    method forward (line 621) | def forward(
  class UpBlock3D (line 670) | class UpBlock3D(nn.Module):
    method __init__ (line 671) | def __init__(
    method forward (line 735) | def forward(self, hidden_states, res_hidden_states_tuple, temb=None, u...

FILE: animatediff/pipelines/pipeline_animation.py
  class AnimationPipelineOutput (line 38) | class AnimationPipelineOutput(BaseOutput):
  class AnimationPipeline (line 42) | class AnimationPipeline(DiffusionPipeline):
    method __init__ (line 45) | def __init__(
    method enable_vae_slicing (line 121) | def enable_vae_slicing(self):
    method disable_vae_slicing (line 124) | def disable_vae_slicing(self):
    method enable_sequential_cpu_offload (line 127) | def enable_sequential_cpu_offload(self, gpu_id=0):
    method _execution_device (line 141) | def _execution_device(self):
    method _encode_prompt (line 153) | def _encode_prompt(self, prompt, device, num_videos_per_prompt, do_cla...
    method decode_latents (line 242) | def decode_latents(self, latents):
    method prepare_extra_step_kwargs (line 257) | def prepare_extra_step_kwargs(self, generator, eta):
    method check_inputs (line 274) | def check_inputs(self, prompt, height, width, callback_steps):
    method prepare_latents (line 289) | def prepare_latents(self, batch_size, num_channels_latents, video_leng...
    method __call__ (line 319) | def __call__(

FILE: animatediff/utils/convert_from_ckpt.py
  function shave_segments (line 53) | def shave_segments(path, n_shave_prefix_segments=1):
  function renew_resnet_paths (line 63) | def renew_resnet_paths(old_list, n_shave_prefix_segments=0):
  function renew_vae_resnet_paths (line 85) | def renew_vae_resnet_paths(old_list, n_shave_prefix_segments=0):
  function renew_attention_paths (line 101) | def renew_attention_paths(old_list, n_shave_prefix_segments=0):
  function renew_vae_attention_paths (line 122) | def renew_vae_attention_paths(old_list, n_shave_prefix_segments=0):
  function assign_to_checkpoint (line 152) | def assign_to_checkpoint(
  function conv_attn_to_linear (line 203) | def conv_attn_to_linear(checkpoint):
  function create_unet_diffusers_config (line 215) | def create_unet_diffusers_config(original_config, image_size: int, contr...
  function create_vae_diffusers_config (line 284) | def create_vae_diffusers_config(original_config, image_size: int):
  function create_diffusers_schedular (line 308) | def create_diffusers_schedular(original_config):
  function create_ldm_bert_config (line 318) | def create_ldm_bert_config(original_config):
  function convert_ldm_unet_checkpoint (line 328) | def convert_ldm_unet_checkpoint(checkpoint, config, path=None, extract_e...
  function convert_ldm_vae_checkpoint (line 559) | def convert_ldm_vae_checkpoint(checkpoint, config):
  function convert_ldm_bert_checkpoint (line 666) | def convert_ldm_bert_checkpoint(checkpoint, config):
  function convert_ldm_clip_checkpoint (line 716) | def convert_ldm_clip_checkpoint(checkpoint):
  function convert_paint_by_example_checkpoint (line 755) | def convert_paint_by_example_checkpoint(checkpoint):
  function convert_open_clip_checkpoint (line 822) | def convert_open_clip_checkpoint(checkpoint):
  function stable_unclip_image_encoder (line 865) | def stable_unclip_image_encoder(original_config):
  function stable_unclip_image_noising_components (line 898) | def stable_unclip_image_noising_components(
  function convert_controlnet_checkpoint (line 943) | def convert_controlnet_checkpoint(

FILE: animatediff/utils/convert_lora_safetensor_to_diffusers.py
  function load_diffusers_lora (line 27) | def load_diffusers_lora(pipeline, state_dict, alpha=1.0):
  function convert_lora (line 50) | def convert_lora(pipeline, state_dict, LORA_PREFIX_UNET="lora_unet", LOR...

FILE: animatediff/utils/util.py
  function zero_rank_print (line 57) | def zero_rank_print(s):
  function save_videos_grid (line 61) | def save_videos_grid(videos: torch.Tensor, path: str, rescale=False, n_r...
  function auto_download (line 76) | def auto_download(local_path, is_dreambooth_lora=False):
  function load_weights (line 91) | def load_weights(

FILE: app.py
  class AnimateController (line 54) | class AnimateController:
    method __init__ (line 55) | def __init__(self):
    method refresh_stable_diffusion (line 84) | def refresh_stable_diffusion(self):
    method refresh_personalized_model (line 87) | def refresh_personalized_model(self):
    method update_pipeline (line 92) | def update_pipeline(
    method update_pipeline_alpha (line 145) | def update_pipeline_alpha(
    method animate (line 170) | def animate(
  function ui (line 224) | def ui():

FILE: scripts/animate.py
  function main (line 32) | def main(args):

FILE: train.py
  function init_dist (line 44) | def init_dist(launcher="slurm", backend='nccl', port=29500, **kwargs):
  function main (line 77) | def main(