Functions — github.com/MatrixTeam-AI/RAIN

CodeHub MCPcopy Create free account

hub / github.com/MatrixTeam-AI/RAIN / functions

Functions352 in github.com/MatrixTeam-AI/RAIN

⨍ Functions 352◇ Types & classes 79

Methodattn_processors

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

src/models/unet_3d_explicit_reference.py:278

Methodattn_processors

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

src/models/unet_3d.py:278

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

Functionchange_cropping

gradio_app.py:94

Functionchange_fps

gradio_app.py:91

Functionclear_queue

gradio_app.py:87

Functionclear_ref

gradio_app.py:81

Methodcreate_custom_forward

(module, return_dict=None)

src/models/transformer_3d.py:149

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

src/models/unet_2d_blocks.py:748

Methodcreate_custom_forward

(module, return_dict=None)

src/models/unet_2d_blocks.py:916

Methodcreate_custom_forward

src/models/unet_2d_blocks.py:1050

Methodcreate_custom_forward

(module, return_dict=None)

src/models/unet_3d_blocks.py:445

Methodcreate_custom_forward

src/models/unet_3d_blocks.py:584

Methodcreate_custom_forward

(module, return_dict=None)

src/models/unet_3d_blocks.py:751

Methodcreate_custom_forward

src/models/unet_3d_blocks.py:888

Methodcustom_forward

src/models/transformer_3d.py:150

Methodcustom_forward

src/models/transformer_2d.py:334

Methodcustom_forward

src/models/unet_2d_blocks.py:471

Methodcustom_forward

src/models/unet_2d_blocks.py:630

Methodcustom_forward

src/models/unet_2d_blocks.py:749

Methodcustom_forward

src/models/unet_2d_blocks.py:917

Methodcustom_forward

src/models/unet_2d_blocks.py:1051

Methodcustom_forward

src/models/unet_3d_blocks.py:446

Methodcustom_forward

src/models/unet_3d_blocks.py:585

Methodcustom_forward

src/models/unet_3d_blocks.py:752

Methodcustom_forward

src/models/unet_3d_blocks.py:889

Methoddecode_slice

(self, x, pmems)

src/taesdv/taesdv.py:143

src/dwpose/__init__.py:101

Methoddisable_freeu

Disables the FreeU mechanism.

src/models/unet_2d_condition.py:861

Methoddisable_vae_slicing

src/pipeline/pipeline_pose2vid_lcm.py:76

Methoddisable_vae_slicing

src/pipeline/pipeline_pose2vid.py:76

Functiondraw_handpose

(canvas, all_hand_peaks)

src/dwpose/util.py:112

Functiondraw_pose

src/dwpose/util.py:140

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/pipeline/pipeline_pose2vid_lcm.py:79

Methodenable_sequential_cpu_offload

(self, gpu_id=0)

src/pipeline/pipeline_pose2vid.py:79

Methodenable_vae_slicing

src/pipeline/pipeline_pose2vid_lcm.py:73

Methodenable_vae_slicing

src/pipeline/pipeline_pose2vid.py:73

Methodenable_xformers_memory_efficient_attention

src/morpher.py:413

Methodextra_repr

src/models/motion_module.py:306

FunctionfaceDetect

(candidate, subset, oriImg)

src/dwpose/util.py:268

Methodfn_recursive_add_processors

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

src/models/unet_3d_explicit_reference.py:287

Methodfn_recursive_add_processors

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

src/models/unet_3d.py:287

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_explicit_reference.py:406

Methodfn_recursive_attn_processor

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

src/models/unet_3d.py:406

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_explicit_reference.py:323

Methodfn_recursive_retrieve_slicable_dims

(module: torch.nn.Module)

src/models/unet_3d.py:323

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_explicit_reference.py:365

Methodfn_recursive_set_attention_slice

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

src/models/unet_3d.py:365

Methodfn_recursive_set_attention_slice

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

src/models/unet_2d_condition.py:820

(self, sample, encoder_hidden_states, pose_cond_fea, pose, add_noise, new_noise, d00, d01, d10, d11,

src/modeling/framed_models.py:37

(self, latents, clip_embeds)

src/modeling/framed_models.py:110

src/modeling/framed_models.py:142

src/modeling/framed_models.py:171

src/taesdv/taesdv.py:16

src/taesdv/taesdv.py:25

src/taesdv/taesdv.py:34

r""" Args: sample (`torch.FloatTensor`): (batch, channel, height, width) noisy inputs tensor timestep (`torch.FloatTen

src/models/unet_3d_explicit_reference.py:421

r""" The forward method of the `Attention` class. Args: hidden_states (`torch.Tensor`): The hidden states

src/models/attention_processor.py:496

(self, f: torch.FloatTensor, zq: torch.FloatTensor)

src/models/attention_processor.py:1698

( self, hidden_states, encoder_hidden_states=None, self_attention_additional_f

src/models/transformer_3d.py:106

The [`Transformer2DModel`] forward method. Args: hidden_states (`torch.LongTensor` of shape `(batch size, num latent pix

src/models/transformer_2d.py:213

r""" Args: sample (`torch.FloatTensor`): (batch, channel, height, width) noisy inputs tensor timestep (`torch.FloatTen

src/models/unet_3d.py:421

r""" The [`UNet2DConditionModel`] forward method. Args: sample (`torch.FloatTensor`): The noisy input ten

src/models/unet_2d_condition.py:872

( self, hidden_states: torch.FloatTensor, attention_mask: Optional[torch.FloatTensor]

src/models/attention.py:179

( self, hidden_states, encoder_hidden_states=None, timestep=None, atte

src/models/attention.py:384

(self, x: torch.FloatTensor)

src/models/unet_2d_blocks.py:219

( self, hidden_states: torch.FloatTensor, temb: Optional[torch.FloatTensor] = None )

src/models/unet_2d_blocks.py:344

( self, hidden_states: torch.FloatTensor, temb: Optional[torch.FloatTensor] = None,

src/models/unet_2d_blocks.py:452

( self, hidden_states: torch.FloatTensor, temb: Optional[torch.FloatTensor] = None,

src/models/unet_2d_blocks.py:606

( self, hidden_states: torch.FloatTensor, temb: Optional[torch.FloatTensor] = None,

src/models/unet_2d_blocks.py:737

( self, hidden_states: torch.FloatTensor, res_hidden_states_tuple: Tuple[torch.FloatTe

src/models/unet_2d_blocks.py:872

( self, hidden_states: torch.FloatTensor, res_hidden_states_tuple: Tuple[torch.FloatTe

src/models/unet_2d_blocks.py:1014

( self, hidden_states, temb=None, encoder_hidden_states=None, attentio

src/models/unet_3d_blocks.py:282

( self, hidden_states, temb=None, encoder_hidden_states=None, attentio

src/models/unet_3d_blocks.py:424

(self, hidden_states, temb=None, encoder_hidden_states=None)

src/models/unet_3d_blocks.py:577

( self, hidden_states, res_hidden_states_tuple, temb=None, encoder_hid

src/models/unet_3d_blocks.py:726

( self, hidden_states, res_hidden_states_tuple, temb=None, upsample_si

src/models/unet_3d_blocks.py:871

(self, conditioning)

src/models/pose_guider.py:47

src/models/resnet.py:10

src/models/resnet.py:21

(self, hidden_states, output_size=None)

src/models/resnet.py:53

(self, hidden_states)

src/models/resnet.py:112

(self, input_tensor, temb)

src/models/resnet.py:217

( self, input_tensor, temb, encoder_hidden_states, attention_mask=None

src/models/motion_module.py:77

(self, hidden_states, encoder_hidden_states=None, attention_mask=None)

src/models/motion_module.py:146

( self, hidden_states, encoder_hidden_states=None, attention_mask=None,

src/models/motion_module.py:236

src/models/motion_module.py:275

( self, hidden_states, encoder_hidden_states=None, attention_mask=None,

src/models/motion_module.py:351

Methodfrom_pretrained_2d

( cls, pretrained_model_path: PathLike, motion_module_path=None, subfolder=Non

src/models/unet_3d_explicit_reference.py:648

gradio_app.py:133

Functionfuse_ref

gradio_app.py:65

Functionget_down_block

( down_block_type, num_layers, in_channels, out_channels, temb_channels, add_downsampl

src/models/unet_3d_blocks.py:15

Methodget_dynamic_axes

src/modeling/framed_models.py:134

← previous next →201–300 of 352, ranked by callers