Functions — github.com/MooreThreads/Moore-AnimateAnyone

Method__init__

( self, img_size, img_scale=(1.0, 1.0), img_ratio=(0.9, 1.0), drop_rat

src/dataset/dance_image.py:13

Method__init__

( self, sample_rate, n_sample_frames, width, height, img_scale

src/dataset/dance_video.py:16

Method__init__

( self, vae, image_encoder, reference_unet, denoising_unet, po

src/pipelines/pipeline_lmks2vid_long.py:56

Method__init__

( self, vae, image_encoder, reference_unet, denoising_unet, po

src/pipelines/pipeline_pose2vid.py:30

Method__init__

( self, vae, image_encoder, reference_unet, denoising_unet, po

src/pipelines/pipeline_pose2img.py:34

Method__init__

( self, vae, image_encoder, reference_unet, denoising_unet, po

src/pipelines/pipeline_pose2vid_long.py:38

Method__init__

(self, device="cuda:0")

src/dwpose/wholebody.py:15

Method__init__

(self)

src/dwpose/__init__.py:40

Method__init__

( self, num_attention_heads: int = 16, attention_head_dim: int = 88, in_channe

src/models/transformer_3d.py:31

Method__init__

( self, unet, mode="write", do_classifier_free_guidance=False, attenti

src/models/mutual_self_attention.py:20

Method__init__

( self, num_attention_heads: int = 16, attention_head_dim: int = 88, in_channe

src/models/transformer_2d.py:63

Method__init__

( self, sample_size: Optional[int] = None, in_channels: int = 4, out_channels:

src/models/unet_3d.py:37

Method__init__

( self, sample_size: Optional[int] = None, in_channels: int = 4, out_channels:

src/models/unet_2d_condition.py:161

Method__init__

( self, dim: int, num_attention_heads: int, attention_head_dim: int, d

src/models/attention.py:299

Method__init__

(self, in_channels: int, out_channels: int, act_fn: str)

src/models/unet_2d_blocks.py:202

Method__init__

( self, in_channels: int, temb_channels: int, dropout: float = 0.0, nu

src/models/unet_2d_blocks.py:254

Method__init__

( self, in_channels: int, temb_channels: int, dropout: float = 0.0, nu

src/models/unet_2d_blocks.py:357

Method__init__

( self, in_channels: int, out_channels: int, temb_channels: int, dropo

src/models/unet_2d_blocks.py:511

Method__init__

( self, in_channels: int, out_channels: int, temb_channels: int, dropo

src/models/unet_2d_blocks.py:682

Method__init__

( self, in_channels: int, out_channels: int, prev_output_channel: int,

src/models/unet_2d_blocks.py:780

Method__init__

( self, in_channels: int, temb_channels: int, dropout: float = 0.0, nu

src/models/unet_3d_blocks.py:180

Method__init__

( self, in_channels: int, out_channels: int, temb_channels: int, dropo

src/models/unet_3d_blocks.py:314

Method__init__

( self, in_channels: int, out_channels: int, temb_channels: int, dropo

src/models/unet_3d_blocks.py:498

Method__init__

( self, in_channels: int, out_channels: int, prev_output_channel: int,

src/models/unet_3d_blocks.py:617

Method__init__

( self, conditioning_embedding_channels: int, conditioning_channels: int = 3,

src/models/pose_guider.py:13

Method__init__

( self, channels, use_conv=False, out_channels=None, padding=1, name="conv" )

src/models/resnet.py:94

Method__init__

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

src/models/resnet.py:124

Method__init__

( self, in_channels, num_attention_heads=8, num_transformer_block=2, a

src/models/motion_module.py:45

Method__init__

( self, in_channels, num_attention_heads, attention_head_dim, num_laye

src/models/motion_module.py:95

Method__init__

( self, dim, num_attention_heads, attention_head_dim, attention_block_

src/models/motion_module.py:186

Method__init__

( self, attention_mode=None, cross_frame_attention_mode=None, temporal_positio

src/models/motion_module.py:281

Method__len__

(self)

src/dataset/dance_image.py:123

Method__len__

(self)

src/dataset/dance_video.py:136

Method_encode_prompt

( self, prompt, device, num_videos_per_prompt, do_classifier_free_guid

src/pipelines/pipeline_pose2vid.py:176

Method_encode_prompt

( self, prompt, device, num_videos_per_prompt, do_classifier_free_guid

src/pipelines/pipeline_pose2vid_long.py:184

Method_execution_device

(self)

src/pipelines/pipeline_lmks2vid_long.py:122

Method_execution_device

(self)

src/pipelines/pipeline_pose2vid.py:92

Method_execution_device

(self)

src/pipelines/pipeline_pose2img.py:90

Method_execution_device

(self)

src/pipelines/pipeline_pose2vid_long.py:100

Method_set_gradient_checkpointing

(self, module, value=False)

src/models/transformer_3d.py:102

Method_set_gradient_checkpointing

(self, module, value=False)

src/models/transformer_2d.py:209

Method_set_gradient_checkpointing

(self, module, value=False)

src/models/unet_3d.py:374

Method_set_gradient_checkpointing

(self, module, value=False)

src/models/unet_2d_condition.py:833

Methodanimate

( self, ref_image, pose_video_path, width=512, height=768, len

app.py:33

Methodattn_processors

r""" Returns: `dict` of attention processors: A dictionary containing all attention processors used in the model with

src/models/unet_3d.py:274

Methodattn_processors

r""" Returns: `dict` of attention processors: A dictionary containing all attention processors used in the model with

src/models/unet_2d_condition.py:672

Methodcal_height

(self, input_image)

src/dwpose/__init__.py:47

Methodcreate_custom_forward

(module, return_dict=None)

src/models/transformer_3d.py:148

Methodcreate_custom_forward

(module, return_dict=None)

src/models/transformer_2d.py:333

Methodcreate_custom_forward

(module, return_dict=None)

src/models/unet_2d_blocks.py:470

Methodcreate_custom_forward

(module, return_dict=None)

src/models/unet_2d_blocks.py:629

Methodcreate_custom_forward

(module)

src/models/unet_2d_blocks.py:748

Methodcreate_custom_forward

(module, return_dict=None)

src/models/unet_2d_blocks.py:916

Methodcreate_custom_forward

(module)

src/models/unet_2d_blocks.py:1050

Methodcreate_custom_forward

(module, return_dict=None)

src/models/unet_3d_blocks.py:439

Methodcreate_custom_forward

(module)

src/models/unet_3d_blocks.py:577

Methodcreate_custom_forward

(module, return_dict=None)

src/models/unet_3d_blocks.py:740

Methodcreate_custom_forward

(module)

src/models/unet_3d_blocks.py:876

Methodcustom_forward

(*inputs)

src/models/transformer_3d.py:149

Methodcustom_forward

(*inputs)

src/models/transformer_2d.py:334

Methodcustom_forward

(*inputs)

src/models/unet_2d_blocks.py:471

Methodcustom_forward

(*inputs)

src/models/unet_2d_blocks.py:630

Methodcustom_forward

(*inputs)

src/models/unet_2d_blocks.py:749

Methodcustom_forward

(*inputs)

src/models/unet_2d_blocks.py:917

Methodcustom_forward

(*inputs)

src/models/unet_2d_blocks.py:1051

Methodcustom_forward

(*inputs)

src/models/unet_3d_blocks.py:440

Methodcustom_forward

(*inputs)

src/models/unet_3d_blocks.py:578

Methodcustom_forward

(*inputs)

src/models/unet_3d_blocks.py:741

Methodcustom_forward

(*inputs)

src/models/unet_3d_blocks.py:877

Functiondecode_latents

(vae, latents)

train_stage_2.py:746

Methoddisable_freeu

Disables the FreeU mechanism.

src/models/unet_2d_condition.py:861

Methoddisable_vae_slicing

(self)

src/pipelines/pipeline_lmks2vid_long.py:106

Methoddisable_vae_slicing

(self)

src/pipelines/pipeline_pose2vid.py:76

Methoddisable_vae_slicing

(self)

src/pipelines/pipeline_pose2img.py:74

Methoddisable_vae_slicing

(self)

src/pipelines/pipeline_pose2vid_long.py:84

Functiondraw_bodypose

(canvas, candidate, subset)

src/dwpose/util.py:79

Functiondraw_facepose

(canvas, all_lmks)

src/dwpose/util.py:215

Functiondraw_handpose

(canvas, all_hand_peaks)

src/dwpose/util.py:160

Methodenable_freeu

r"""Enables the FreeU mechanism from https://arxiv.org/abs/2309.11497. The suffixes after the scaling factors represent the stage blocks wher

src/models/unet_2d_condition.py:837

Methodenable_sequential_cpu_offload

(self, gpu_id=0)

src/pipelines/pipeline_lmks2vid_long.py:109

Methodenable_sequential_cpu_offload

(self, gpu_id=0)

src/pipelines/pipeline_pose2vid.py:79

Methodenable_sequential_cpu_offload

(self, gpu_id=0)

src/pipelines/pipeline_pose2img.py:77

Methodenable_sequential_cpu_offload

(self, gpu_id=0)

src/pipelines/pipeline_pose2vid_long.py:87

Methodenable_vae_slicing

(self)

src/pipelines/pipeline_lmks2vid_long.py:103

Methodenable_vae_slicing

(self)

src/pipelines/pipeline_pose2vid.py:73

Methodenable_vae_slicing

(self)

src/pipelines/pipeline_pose2img.py:71

Methodenable_vae_slicing

(self)

src/pipelines/pipeline_pose2vid_long.py:81

Methodextra_repr

(self)

src/models/motion_module.py:306

FunctionfaceDetect

(candidate, subset, oriImg)

src/dwpose/util.py:302

Methodfn_recursive_add_processors

( name: str, module: torch.nn.Module, processors: Dict[str, AttentionProce

src/models/unet_3d.py:283

Methodfn_recursive_add_processors

( name: str, module: torch.nn.Module, processors: Dict[str, AttentionProce

src/models/unet_2d_condition.py:681

Methodfn_recursive_attn_processor

(name: str, module: torch.nn.Module, processor)

src/models/unet_3d.py:402

Methodfn_recursive_attn_processor

(name: str, module: torch.nn.Module, processor)

src/models/unet_2d_condition.py:726

Methodfn_recursive_retrieve_slicable_dims

(module: torch.nn.Module)

src/models/unet_3d.py:319

Methodfn_recursive_retrieve_sliceable_dims

(module: torch.nn.Module)

src/models/unet_2d_condition.py:778

Methodfn_recursive_set_attention_slice

( module: torch.nn.Module, slice_size: List[int] )

src/models/unet_3d.py:361

Methodfn_recursive_set_attention_slice

( module: torch.nn.Module, slice_size: List[int] )

src/models/unet_2d_condition.py:820

Methodforward

( self, noisy_latents, timesteps, ref_image_latents, clip_image_embeds

train_stage_2.py:74

Methodforward

( self, noisy_latents, timesteps, ref_image_latents, clip_image_embeds

train_stage_1.py:65

Methodforward

( self, hidden_states, encoder_hidden_states=None, self_attention_additional_f

src/models/transformer_3d.py:106

Functions250 in github.com/MooreThreads/Moore-AnimateAnyone