Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/TMElyralab/MusePose
/ functions
Functions
214 in github.com/TMElyralab/MusePose
⨍
Functions
214
◇
Types & classes
48
Method
__init__
( self, *, in_channels, out_channels=None, conv_shortcut=False,
musepose/models/resnet.py:124
Method
__init__
( self, in_channels, num_attention_heads=8, num_transformer_block=2, a
musepose/models/motion_module.py:45
Method
__init__
( self, in_channels, num_attention_heads, attention_head_dim, num_laye
musepose/models/motion_module.py:95
Method
__init__
( self, dim, num_attention_heads, attention_head_dim, attention_block_
musepose/models/motion_module.py:186
Method
__init__
( self, attention_mode=None, cross_frame_attention_mode=None, temporal_positio
musepose/models/motion_module.py:281
Method
__init__
(self, det_config=None, det_ckpt=None, pose_config=None, pose_ckpt=None, device="cpu", keypoints_only=False)
pose/script/dwpose.py:71
Method
__init__
(self, det_config=None, det_ckpt=None, pose_config=None, pose_ckpt=None,
pose/script/wholebody.py:33
Method
__len__
(self)
musepose/dataset/dance_image.py:129
Method
__len__
(self)
musepose/dataset/dance_video.py:149
Method
_encode_prompt
( self, prompt, device, num_videos_per_prompt, do_classifier_free_guid
musepose/pipelines/pipeline_pose2vid.py:176
Method
_encode_prompt
( self, prompt, device, num_videos_per_prompt, do_classifier_free_guid
musepose/pipelines/pipeline_pose2vid_long.py:184
Method
_execution_device
(self)
musepose/pipelines/pipeline_pose2vid.py:92
Method
_execution_device
(self)
musepose/pipelines/pipeline_pose2img.py:90
Method
_execution_device
(self)
musepose/pipelines/pipeline_pose2vid_long.py:100
Method
_set_gradient_checkpointing
(self, module, value=False)
musepose/models/transformer_3d.py:99
Method
_set_gradient_checkpointing
(self, module, value=False)
musepose/models/transformer_2d.py:208
Method
_set_gradient_checkpointing
(self, module, value=False)
musepose/models/unet_3d.py:354
Method
_set_gradient_checkpointing
(self, module, value=False)
musepose/models/unet_2d_condition.py:832
Method
attn_processors
r""" Returns: `dict` of attention processors: A dictionary containing all attention processors used in the model with
musepose/models/unet_3d.py:254
Method
attn_processors
r""" Returns: `dict` of attention processors: A dictionary containing all attention processors used in the model with
musepose/models/unet_2d_condition.py:671
Method
create_custom_forward
(module, return_dict=None)
musepose/models/transformer_2d.py:332
Method
create_custom_forward
(module, return_dict=None)
musepose/models/unet_2d_blocks.py:470
Method
create_custom_forward
(module, return_dict=None)
musepose/models/unet_2d_blocks.py:629
Method
create_custom_forward
(module)
musepose/models/unet_2d_blocks.py:748
Method
create_custom_forward
(module, return_dict=None)
musepose/models/unet_2d_blocks.py:916
Method
create_custom_forward
(module)
musepose/models/unet_2d_blocks.py:1050
Method
create_custom_forward
(module, return_dict=None)
musepose/models/unet_3d_blocks.py:413
Method
create_custom_forward
(module)
musepose/models/unet_3d_blocks.py:556
Method
create_custom_forward
(module, return_dict=None)
musepose/models/unet_3d_blocks.py:710
Method
create_custom_forward
(module)
musepose/models/unet_3d_blocks.py:841
Method
custom_forward
(*inputs)
musepose/models/transformer_2d.py:333
Method
custom_forward
(*inputs)
musepose/models/unet_2d_blocks.py:471
Method
custom_forward
(*inputs)
musepose/models/unet_2d_blocks.py:630
Method
custom_forward
(*inputs)
musepose/models/unet_2d_blocks.py:749
Method
custom_forward
(*inputs)
musepose/models/unet_2d_blocks.py:917
Method
custom_forward
(*inputs)
musepose/models/unet_2d_blocks.py:1051
Method
custom_forward
(*inputs)
musepose/models/unet_3d_blocks.py:414
Method
custom_forward
(*inputs)
musepose/models/unet_3d_blocks.py:557
Method
custom_forward
(*inputs)
musepose/models/unet_3d_blocks.py:711
Method
custom_forward
(*inputs)
musepose/models/unet_3d_blocks.py:842
Function
decode_latents
(vae, latents)
train_stage_2_multiGPU.py:620
Function
delete_additional_ckpt
(base_path, num_keep)
musepose/utils/util.py:35
Function
delete_additional_ckpt
(base_path, num_keep)
pose/script/tool.py:35
Method
disable_freeu
Disables the FreeU mechanism.
musepose/models/unet_2d_condition.py:860
Method
disable_vae_slicing
(self)
musepose/pipelines/pipeline_pose2vid.py:76
Method
disable_vae_slicing
(self)
musepose/pipelines/pipeline_pose2img.py:74
Method
disable_vae_slicing
(self)
musepose/pipelines/pipeline_pose2vid_long.py:84
Function
draw_bodypose
(canvas, candidate, subset)
pose/script/util.py:28
Function
draw_facepose
(canvas, all_lmks)
pose/script/util.py:111
Function
draw_handpose
(canvas, all_hand_peaks)
pose/script/util.py:73
Method
enable_freeu
r"""Enables the FreeU mechanism from https://arxiv.org/abs/2309.11497. The suffixes after the scaling factors represent the stage blocks wher
musepose/models/unet_2d_condition.py:836
Method
enable_sequential_cpu_offload
(self, gpu_id=0)
musepose/pipelines/pipeline_pose2vid.py:79
Method
enable_sequential_cpu_offload
(self, gpu_id=0)
musepose/pipelines/pipeline_pose2img.py:77
Method
enable_sequential_cpu_offload
(self, gpu_id=0)
musepose/pipelines/pipeline_pose2vid_long.py:87
Method
enable_vae_slicing
(self)
musepose/pipelines/pipeline_pose2vid.py:73
Method
enable_vae_slicing
(self)
musepose/pipelines/pipeline_pose2img.py:71
Method
enable_vae_slicing
(self)
musepose/pipelines/pipeline_pose2vid_long.py:81
Method
extra_repr
(self)
musepose/models/motion_module.py:306
Method
fn_recursive_add_processors
( name: str, module: torch.nn.Module, processors: Dict[str, AttentionProce
musepose/models/unet_3d.py:263
Method
fn_recursive_add_processors
( name: str, module: torch.nn.Module, processors: Dict[str, AttentionProce
musepose/models/unet_2d_condition.py:680
Method
fn_recursive_attn_processor
(name: str, module: torch.nn.Module, processor)
musepose/models/unet_3d.py:382
Method
fn_recursive_attn_processor
(name: str, module: torch.nn.Module, processor)
musepose/models/unet_2d_condition.py:725
Method
fn_recursive_retrieve_slicable_dims
(module: torch.nn.Module)
musepose/models/unet_3d.py:299
Method
fn_recursive_retrieve_sliceable_dims
(module: torch.nn.Module)
musepose/models/unet_2d_condition.py:777
Method
fn_recursive_set_attention_slice
( module: torch.nn.Module, slice_size: List[int] )
musepose/models/unet_3d.py:341
Method
fn_recursive_set_attention_slice
( module: torch.nn.Module, slice_size: List[int] )
musepose/models/unet_2d_condition.py:819
Method
forward
( self, noisy_latents, timesteps, ref_image_latents, clip_image_embeds
train_stage_1_multiGPU.py:69
Method
forward
( self, noisy_latents, timesteps, ref_image_latents, clip_image_embeds
train_stage_2_multiGPU.py:76
Method
forward
( self, hidden_states, encoder_hidden_states=None, timestep=None, retu
musepose/models/transformer_3d.py:103
Method
forward
The [`Transformer2DModel`] forward method. Args: hidden_states (`torch.LongTensor` of shape `(batch size, num latent pix
musepose/models/transformer_2d.py:212
Method
forward
r""" Args: sample (`torch.FloatTensor`): (batch, channel, height, width) noisy inputs tensor timestep (`torch.FloatTen
musepose/models/unet_3d.py:397
Method
forward
r""" The [`UNet2DConditionModel`] forward method. Args: sample (`torch.FloatTensor`): The noisy input ten
musepose/models/unet_2d_condition.py:871
Method
forward
( self, hidden_states: torch.FloatTensor, attention_mask: Optional[torch.FloatTensor]
musepose/models/attention.py:178
Method
forward
( self, hidden_states, encoder_hidden_states=None, timestep=None, atte
musepose/models/attention.py:381
Method
forward
(self, x: torch.FloatTensor)
musepose/models/unet_2d_blocks.py:219
Method
forward
( self, hidden_states: torch.FloatTensor, temb: Optional[torch.FloatTensor] = None )
musepose/models/unet_2d_blocks.py:344
Method
forward
( self, hidden_states: torch.FloatTensor, temb: Optional[torch.FloatTensor] = None,
musepose/models/unet_2d_blocks.py:452
Method
forward
( self, hidden_states: torch.FloatTensor, temb: Optional[torch.FloatTensor] = None,
musepose/models/unet_2d_blocks.py:606
Method
forward
( self, hidden_states: torch.FloatTensor, temb: Optional[torch.FloatTensor] = None,
musepose/models/unet_2d_blocks.py:737
Method
forward
( self, hidden_states: torch.FloatTensor, res_hidden_states_tuple: Tuple[torch.FloatTe
musepose/models/unet_2d_blocks.py:872
Method
forward
( self, hidden_states: torch.FloatTensor, res_hidden_states_tuple: Tuple[torch.FloatTe
musepose/models/unet_2d_blocks.py:1014
Method
forward
( self, hidden_states, temb=None, encoder_hidden_states=None, attentio
musepose/models/unet_3d_blocks.py:269
Method
forward
( self, hidden_states, temb=None, encoder_hidden_states=None, attentio
musepose/models/unet_3d_blocks.py:398
Method
forward
(self, hidden_states, temb=None, encoder_hidden_states=None)
musepose/models/unet_3d_blocks.py:549
Method
forward
( self, hidden_states, res_hidden_states_tuple, temb=None, encoder_hid
musepose/models/unet_3d_blocks.py:691
Method
forward
( self, hidden_states, res_hidden_states_tuple, temb=None, upsample_si
musepose/models/unet_3d_blocks.py:824
Method
forward
(self, conditioning)
musepose/models/pose_guider.py:47
Method
forward
(self, x)
musepose/models/resnet.py:10
Method
forward
(self, x)
musepose/models/resnet.py:21
Method
forward
(self, hidden_states, output_size=None)
musepose/models/resnet.py:53
Method
forward
(self, hidden_states)
musepose/models/resnet.py:112
Method
forward
(self, input_tensor, temb)
musepose/models/resnet.py:217
Method
forward
( self, input_tensor, temb, encoder_hidden_states, attention_mask=None
musepose/models/motion_module.py:77
Method
forward
(self, hidden_states, encoder_hidden_states=None, attention_mask=None)
musepose/models/motion_module.py:146
Method
forward
( self, hidden_states, encoder_hidden_states=None, attention_mask=None,
musepose/models/motion_module.py:236
Method
forward
(self, x)
musepose/models/motion_module.py:275
Method
forward
( self, hidden_states, encoder_hidden_states=None, attention_mask=None,
musepose/models/motion_module.py:351
Function
get_down_block
( down_block_type, num_layers, in_channels, out_channels, temb_channels, add_downsampl
musepose/models/unet_3d_blocks.py:15
Function
get_fps
(video_path)
pose/script/tool.py:125
Function
get_total_steps
( scheduler, timesteps: List[int], num_steps: Optional[int] = None, num_frames: int = ...,
musepose/pipelines/context.py:52
← previous
next →
101–200 of 214, ranked by callers