Functions — github.com/MatrixTeam-AI/RAIN

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

Functions352 in github.com/MatrixTeam-AI/RAIN

⨍ Functions 352◇ Types & classes 79

Methodget_dynamic_axes

(self)

src/modeling/framed_models.py:163

Methodget_dynamic_axes

(self)

src/modeling/framed_models.py:192

Methodget_input_names

(self)

src/modeling/framed_models.py:128

Methodget_input_names

(self)

src/modeling/framed_models.py:157

Methodget_input_names

(self)

src/modeling/framed_models.py:186

Methodget_output_names

(self)

src/modeling/framed_models.py:131

Methodget_output_names

(self)

src/modeling/framed_models.py:160

Methodget_output_names

(self)

src/modeling/framed_models.py:189

Methodget_sample_input

(self, batchsize, height, width, dtype, device)

src/modeling/framed_models.py:115

Methodget_sample_input

(self, batchsize, height, width, dtype, device)

src/modeling/framed_models.py:145

Methodget_sample_input

(self, batchsize, height, width, dtype, device)

src/modeling/framed_models.py:174

Functionget_up_block

( up_block_type, num_layers, in_channels, out_channels, prev_output_channel, temb_chan

src/models/unet_3d_blocks.py:98

Methodget_velocity

( self, sample: torch.FloatTensor, noise: torch.FloatTensor, timesteps: torch.IntTensor )

src/scheduler/scheduler_lcm.py:511

Methodguidance_scale_embedding

(w, embedding_dim=512, dtype=torch.float32)

src/morpher.py:66

Methodguidance_scale_embedding

(w, embedding_dim=512, dtype=torch.float32)

src/pipeline/pipeline_pose2vid_lcm.py:334

Methodhacked_basic_transformer_inner_forward

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

src/models/mutual_self_attention.py:98

FunctionhandDetect

(candidate, subset, oriImg)

src/dwpose/util.py:202

Functionhandle_onnx_batch_norm

(onnx_path: str)

src/modeling/onnx_export.py:98

Methodimgmxn

(self, imgs, m, n=None, size = None, centercrop=False)

src/morpher.py:749

Methodlatentsmxn

(self, latents, m, n=None, size=None, centercrop=False)

src/morpher.py:762

Methodlink

(self, other, var_map, skip_check=SKIP_ENGINE_MODEL_CHECK)

src/modeling/engine_model.py:206

Functionnpmax

(array)

src/dwpose/util.py:339

FunctionpadRightDownCorner

(img, stride, padValue)

src/dwpose/util.py:38

Methodprevious_timestep

(self, timestep)

src/scheduler/scheduler_lcm.py:535

(self, imgs, vid_file_name, fps=24)

src/morpher.py:709

Methodscale_model_input

Ensures interchangeability with schedulers that need to scale the denoising model input depending on the current timestep. A

src/scheduler/scheduler_lcm.py:266

Methodseed

(self, x)

src/morpher.py:410

Methodset_attention_slice

r""" Set the slice size for attention computation. Args: slice_size (`int`): The slice size for attention

src/models/attention_processor.py:343

Methodset_attn_processor

r""" Sets the attention processor to use to compute attention. Parameters: processor (`dict` of `AttentionProcessor` or o

src/models/unet_3d_explicit_reference.py:383

Methodset_attn_processor

r""" Sets the attention processor to use to compute attention. Parameters: processor (`dict` of `AttentionProcessor` or o

src/models/unet_3d.py:383

Methodset_chunk_feed_forward

(self, chunk_size: Optional[int], dim: int = 0)

src/models/attention.py:174

Methodset_default_attn_processor

Disables custom attention processors and sets the default attention implementation.

src/models/unet_2d_condition.py:741

Methodset_patterns

(self, patterns)

src/dwpose/__init__.py:51

Functionset_translations

(frame)

gradio_app.py:121

Methodset_translations

(self, translations)

src/dwpose/__init__.py:48

Methodset_use_memory_efficient_attention_xformers

r""" Set whether to use memory efficient attention from `xformers` or not. Args: use_memory_efficient_attention_xformers

src/models/attention_processor.py:209

Methodset_use_memory_efficient_attention_xformers

( self, use_memory_efficient_attention_xformers: bool, attention_op: Optional[Callable

src/models/motion_module.py:309

Functionsmart_resize

(x, s)

src/dwpose/util.py:10

Functionsmart_resize_k

(x, fx, fy)

src/dwpose/util.py:24

Functionstart_offline_process

(vid)

gradio_app.py:150

Functionstart_process

src/scheduler/scheduler_lcm.py:263

Functionstop_offline_process

(model, model_weights)

src/dwpose/util.py:61

Functiontransform_func

src/modeling/engine_model.py:261

(self, vids, m, n=None, size=None, length=None, centercrop=True)

src/morpher.py:689

← previous301–352 of 352, ranked by callers