Functions — github.com/alibaba/animate-anything

hub / github.com/alibaba/animate-anything / functions

Functions387 in github.com/alibaba/animate-anything

⨍ Functions 387◇ Types & classes 40

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

models/unet_3d_condition_mask.py:338

Methodforward

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

models/unet_3d_blocks.py:335

Methodforward

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

models/unet_3d_blocks.py:484

Methodforward

(self, hidden_states, temb=None, num_frames=1)

models/unet_3d_blocks.py:599

Methodforward

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

models/unet_3d_blocks.py:713

Methodforward

(self, hidden_states, res_hidden_states_tuple, temb=None, upsample_size=None, num_frames=1)

models/unet_3d_blocks.py:823

Functiongenerate_center_mask

(image)

utils/common.py:223

Functiongenerate_random_mask

(image)

utils/common.py:202

Methodget_average_attention

(self)

utils/ptp_utils.py:236

Methodget_frame_batch

(self, vr, resize=None)

utils/dataset.py:297

Methodget_frame_batch

(self, vr, resize=None)

utils/dataset.py:476

Methodget_frame_buckets

(self, vr)

utils/dataset.py:197

Methodget_frame_buckets

(self, vr)

utils/dataset.py:305

Methodget_frame_buckets

(self, vr)

utils/dataset.py:469

Methodget_frame_buckets

(self, vr)

utils/dataset.py:568

Functionget_full_white_area_mask

(frames)

utils/common.py:81

Methodget_images_list

(self, image_dir)

utils/dataset.py:388

Functionget_optimizer

(use_8bit_adam)

train_transparent_i2v_stage2.py:250

Functionget_refinement_mapper

(prompts, tokenizer, max_len=77)

utils/seq_aligner.py:121

Functionget_replacement_mapper

(prompts, tokenizer, max_len=77)

utils/seq_aligner.py:189

Functionget_text_prompt

( text_prompt: str = '', fallback_prompt: str= '', file_path:str = '', ext_t

utils/dataset.py:61

Functionget_time_words_attention_alpha

(prompts, num_steps, cross_replace_steps: Union[float, Dict[str, Tuple[floa

utils/ptp_utils.py:397

Functionhandle_trainable_modules

(model, trainable_modules, not_trainable_modules=[], is_enabled=True, negation=None)

train_transparent_i2v_stage2.py:279

Functioninit_latent

(latent, model, height, width, generator, batch_size)

utils/ptp_utils.py:163

Functioninject_trainable_lora

inject lora into model, and returns lora parameter groups.

utils/lora.py:336

Functioninject_trainable_lora_extended

inject lora into model, and returns lora parameter groups.

Functionis_mixed_precision

(accelerator)

train_transparent_i2v_stage2.py:263

Functionlatent2image

(vae, latents)

utils/ptp_utils.py:154

Functionload_512

(image_path, left=0, right=0, top=0, bottom=0)

utils/ptp_utils.py:252

Functionload_lora

(model, lora_path: str)

stable_lora/lora.py:369

Functionload_safeloras

(path, device="cpu")

utils/lora.py:765

Functionload_safeloras_both

(path, device="cpu")

utils/lora.py:775

Functionload_safeloras_embeds

(path, device="cpu")

utils/lora.py:770

Functionmerge_overlapping_rectangles

Functionmonkeypatch_add_lora

( model, loras, target_replace_module=DEFAULT_TARGET_REPLACE, alpha: float = 1.0, beta: fl

utils/lora.py:1047

Functionmotion_mask_loss

Functionnegate_params

(name, negation)

train_lora.py:184

Functionnegate_params

(name, negation)

train.py:181

Functionnegate_params

(name, negation)

train_transparent_i2v_stage2.py:207

Functionnegate_params

(name, negation)

train_svd.py:167

Methodnum_uncond_att_layers

(self)

utils/ptp_utils.py:185

Functionparam_optim

(model, condition, extra_params=None, is_lora=False, negation=None)

train_transparent_i2v_stage2.py:184

Functionpatch_pipe

( pipe, maybe_unet_path, token: Optional[str] = None, r: int = 4, patch_unet=True, pat

utils/lora.py:1155

Functionprompt_image

(image, processor, encoder)

train_lora.py:345

Functionprompt_image

(image, processor, encoder)

train.py:317

Functionprompt_image

(image, processor, encoder)

train_transparent_i2v_stage2.py:345

Functionprompt_image

(image, processor, encoder)

train_svd.py:337

Functionread_mask

(json_path, label=["mask"])

utils/common.py:237

Methodrealize_as_lora

(self)

utils/lora.py:141

Methodrealize_as_lora

(self)

utils/lora.py:218

Functionregister_attention_control

(model, controller,controller1)

utils/ptp_utils.py:310

Functionreplace_prompt

(prompt, token, wlist)

train_lora.py:339

Functionreplace_prompt

(prompt, token, wlist)

train.py:311

Functionreplace_prompt

(prompt, token, wlist)

train_transparent_i2v_stage2.py:339

Functionreplace_prompt

(prompt, token, wlist)

train_svd.py:331

Methodreset

(self)

utils/ptp_utils.py:204

Methodreset

(self)

utils/ptp_utils.py:241

Functionsample_noise

(latents, noise_strength, use_offset_noise=False)

train_transparent_i2v_stage2.py:323

Functionsample_noise

(latents, noise_strength, use_offset_noise=False)

train_svd.py:252

Functionsave_all

( unet, text_encoder, save_path, placeholder_token_ids=None, placeholder_tokens=None,

utils/lora.py:1254

Methodsave_lora

(model, name, condition, replace_modules, step, save_path)

utils/lora_handler.py:280

Functionsave_lora_as_json

(model, path="./lora.json")

utils/lora.py:584

Functionsave_safeloras

( modelmap: Dict[str, Tuple[nn.Module, Set[str]]] = {}, outpath="./lora.safetensors", )

utils/lora.py:631

Functionset_lora_diag

(model, diag: torch.Tensor)

utils/lora.py:1080

Methodset_selector_from_diag

(self, diag: torch.Tensor)

utils/lora.py:144

Methodset_selector_from_diag

(self, diag: torch.Tensor)

utils/lora.py:221

Functionshould_sample

(global_step, validation_steps, validation_data)

train_transparent_i2v_stage2.py:334

utils/ptp_utils.py:178

Functiontensor_to_transparent_vae_latent

(t, vae)

train_transparent_i2v_stage2.py:314

Functiontext_under_image

(image: np.ndarray, text: str, text_color: Tuple[int, int, int] = (0, 0, 0))

utils/ptp_utils.py:27

Methodtrain

(self, mode: bool = True)

stable_lora/lora.py:171

Functiontune_lora_scale

(model, alpha: float = 1.0)

utils/lora.py:1074

Functionunet_and_text_g_c

(unet, text_encoder, unet_enable, text_enable)

train_transparent_i2v_stage2.py:132

Functionunfreeze

()

stable_lora/lora.py:251

Functionupdate_num_frames

(input_video, num_frames_textbox)

app_svd.py:166

Functionupdate_to

(b=1, bsize=1, tsize=None)

utils/lama.py:40

Methodvalidate_json

(self, base_path, path)

utils/dataset.py:194

Functionview_images

(images, save_name, num_rows=1, offset_ratio=0.02)

utils/ptp_utils.py:82

← previous301–387 of 387, ranked by callers