SYMBOL INDEX (171 symbols across 16 files)

FILE: i2v_video_sample.py
  function main (line 16) | def main(args):

FILE: motionclone/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=...
  class CrossAttention (line 302) | class CrossAttention(nn.Module):
    method __init__ (line 317) | def __init__(
    method reshape_heads_to_batch_dim (line 367) | def reshape_heads_to_batch_dim(self, tensor):
    method reshape_batch_dim_to_heads (line 374) | def reshape_batch_dim_to_heads(self, tensor):
    method set_attention_slice (line 381) | def set_attention_slice(self, slice_size):
    method forward (line 387) | def forward(self, hidden_states, encoder_hidden_states=None, attention...
    method _attention (line 461) | def _attention(self, query, key, value, attention_mask=None):
    method _sliced_attention (line 492) | def _sliced_attention(self, query, key, value, sequence_length, dim, a...
    method _memory_efficient_attention_xformers (line 535) | def _memory_efficient_attention_xformers(self, query, key, value, atte...
    method set_processor (line 544) | def set_processor(self, processor: "AttnProcessor") -> None:
    method get_attention_scores (line 564) | def get_attention_scores(

FILE: motionclone/models/motion_module.py
  function zero_module (line 21) | def zero_module(module):
  class TemporalTransformer3DModelOutput (line 29) | class TemporalTransformer3DModelOutput(BaseOutput):
  function get_motion_module (line 40) | def get_motion_module( # 只能返回VanillaTemporalModule类
  class VanillaTemporalModule (line 51) | class VanillaTemporalModule(nn.Module):
    method __init__ (line 52) | def __init__(
    method forward (line 80) | def forward(self, input_tensor, temb, encoder_hidden_states, attention...
  class TemporalTransformer3DModel (line 88) | class TemporalTransformer3DModel(nn.Module):
    method __init__ (line 89) | def __init__(
    method forward (line 137) | def forward(self, hidden_states, encoder_hidden_states=None, attention...
  class TemporalTransformerBlock (line 164) | class TemporalTransformerBlock(nn.Module):
    method __init__ (line 165) | def __init__(
    method forward (line 213) | def forward(self, hidden_states, encoder_hidden_states=None, attention...
  class PositionalEncoding (line 228) | class PositionalEncoding(nn.Module):
    method __init__ (line 229) | def __init__(
    method forward (line 245) | def forward(self, x):
  class VersatileAttention (line 250) | class VersatileAttention(CrossAttention): # 继承CrossAttention类，不需要在额外写set...
    method __init__ (line 251) | def __init__(
    method extra_repr (line 271) | def extra_repr(self):
    method forward (line 274) | def forward(self, hidden_states, encoder_hidden_states=None, attention...

FILE: motionclone/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 183) | def forward(self, input_tensor, temb):
  class Mish (line 216) | class Mish(torch.nn.Module):
    method forward (line 217) | def forward(self, hidden_states):

FILE: motionclone/models/scheduler.py
  class CustomDDIMScheduler (line 9) | class CustomDDIMScheduler(DDIMScheduler):
    method step (line 11) | def step(

FILE: motionclone/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 590) | def zero_module(module):

FILE: motionclone/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_path, subfolder=None, une...

FILE: motionclone/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: motionclone/pipelines/pipeline_animation.py
  class AnimationPipelineOutput (line 42) | class AnimationPipelineOutput(BaseOutput):
  class AnimationPipeline (line 46) | class AnimationPipeline(DiffusionPipeline):
    method __init__ (line 49) | def __init__(
    method enable_vae_slicing (line 127) | def enable_vae_slicing(self):
    method disable_vae_slicing (line 130) | def disable_vae_slicing(self):
    method enable_sequential_cpu_offload (line 133) | def enable_sequential_cpu_offload(self, gpu_id=0):
    method _execution_device (line 148) | def _execution_device(self):
    method _encode_prompt (line 160) | def _encode_prompt(self, prompt, device, num_videos_per_prompt, do_cla...
    method decode_latents (line 250) | def decode_latents(self, latents):
    method prepare_extra_step_kwargs (line 265) | def prepare_extra_step_kwargs(self, generator, eta):
    method check_inputs (line 282) | def check_inputs(self, prompt, height, width, callback_steps):
    method prepare_latents (line 297) | def prepare_latents(self, batch_size, num_channels_latents, video_leng...
    method __call__ (line 327) | def __call__(

FILE: motionclone/utils/conv_layer.py
  function conv_forward (line 3) | def conv_forward(self):
  function get_conv_feat (line 53) | def get_conv_feat(unet):
  function prep_unet_conv (line 64) | def prep_unet_conv(unet):

FILE: motionclone/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_concise (line 716) | def convert_ldm_clip_checkpoint_concise(checkpoint):
  function convert_ldm_clip_checkpoint (line 725) | def convert_ldm_clip_checkpoint(checkpoint):
  function convert_paint_by_example_checkpoint (line 764) | def convert_paint_by_example_checkpoint(checkpoint):
  function convert_open_clip_checkpoint (line 831) | def convert_open_clip_checkpoint(checkpoint):
  function stable_unclip_image_encoder (line 874) | def stable_unclip_image_encoder(original_config):
  function stable_unclip_image_noising_components (line 907) | def stable_unclip_image_noising_components(
  function convert_controlnet_checkpoint (line 952) | def convert_controlnet_checkpoint(

FILE: motionclone/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: motionclone/utils/motionclone_functions.py
  function add_noise (line 19) | def add_noise(self, timestep, x_0, noise_pred):
  function obtain_motion_representation (line 26) | def obtain_motion_representation(self, generator=None, motion_representa...
  function compute_temp_loss (line 85) | def compute_temp_loss(self, temp_attn_prob_control_dict):
  function sample_video (line 102) | def sample_video(
  function single_step_video (line 173) | def single_step_video(self, noisy_latents, step_index, step_t,  extra_st...
  function get_temp_attn_prob (line 260) | def get_temp_attn_prob(self,index_select=None):
  function schedule_customized_step (line 286) | def schedule_customized_step(
  function schedule_customized_step_candidate (line 413) | def schedule_customized_step_candidate(
  function schedule_set_timesteps (line 520) | def schedule_set_timesteps(self, num_inference_steps: int, guidance_step...
  class UNet3DConditionOutput (line 582) | class UNet3DConditionOutput(BaseOutput):
  function unet_customized_forward (line 585) | def unet_customized_forward(

FILE: motionclone/utils/util.py
  function zero_rank_print (line 83) | def zero_rank_print(s):
  function save_videos_grid (line 87) | def save_videos_grid(videos: torch.Tensor, path: str, rescale=False, n_r...
  function auto_download (line 101) | def auto_download(local_path, is_dreambooth_lora=False):
  function load_weights (line 115) | def load_weights(
  function video_preprocess (line 217) | def video_preprocess(video_path, height, width, video_length, duration=N...
  function set_nested_item (line 245) | def set_nested_item(dataDict, mapList, value):
  function merge_sweep_config (line 257) | def merge_sweep_config(base_config, update):
  function compute_token_merge_indices (line 271) | def compute_token_merge_indices(tokenizer, prompt: str, word: str, word_...
  function extract_data (line 289) | def extract_data(input_string: str) -> list:
  function generate_hash_key (line 314) | def generate_hash_key(image, prompt=""):
  function save_data (line 332) | def save_data(data, folder_path, key):
  function get_data (line 344) | def get_data(folder_path, key):
  function PILtoTensor (line 359) | def PILtoTensor(data: Image.Image) -> torch.Tensor:
  function TensorToPIL (line 363) | def TensorToPIL(data: torch.Tensor) -> Image.Image:
  function load_image (line 367) | def load_image(
  function rescale_noise_cfg (line 410) | def rescale_noise_cfg(noise_cfg, noise_pred_text, guidance_rescale=0.0):
  function _in_step (line 423) | def _in_step(config, step):
  function classify_blocks (line 434) | def classify_blocks(block_list, name):
  function set_all_seed (line 442) | def set_all_seed(seed):

FILE: motionclone/utils/xformer_attention.py
  function identify_blocks (line 8) | def identify_blocks(block_list, name):
  class MySelfAttnProcessor (line 17) | class MySelfAttnProcessor:
    method __init__ (line 18) | def __init__(self, attention_op: Optional[Callable] = None):
    method __call__ (line 22) | def __call__(self, attn, hidden_states, query, key, value, attention_m...
    method record_qkv (line 31) | def record_qkv(self, attn, hidden_states, query, key, value, attention...
    method record_attn_mask (line 40) | def record_attn_mask(self, attn, hidden_states, query, key, value, att...
  function prep_unet_attention (line 45) | def prep_unet_attention(unet,motion_gudiance_blocks):
  function get_self_attn_feat (line 55) | def get_self_attn_feat(unet, injection_config, config):
  function clean_attn_buffer (line 81) | def clean_attn_buffer(unet):

FILE: t2v_video_sample.py
  function main (line 14) | def main(args):