MCPcopy Create free account

hub / github.com/PKU-YuanGroup/MagicTime / functions

Functions144 in github.com/PKU-YuanGroup/MagicTime

⨍ Functions 144◇ Types & classes 26
Method_set_gradient_checkpointing
(self, module, value=False)
utils/unet.py:314
Functionconvert_lora
(pipeline, state_dict, LORA_PREFIX_UNET="lora_unet", LORA_PREFIX_TEXT_ENCODER="lora_te", alpha=0.6)
utils/util.py:197
Methodcreate_custom_forward
(module, return_dict=None)
utils/unet_blocks.py:1177
Methodcreate_custom_forward
(module)
utils/unet_blocks.py:1287
Methodcreate_custom_forward
(module, return_dict=None)
utils/unet_blocks.py:1427
Methodcreate_custom_forward
(module)
utils/unet_blocks.py:1532
Methodcustom_forward
(*inputs)
utils/unet_blocks.py:1178
Methodcustom_forward
(*inputs)
utils/unet_blocks.py:1288
Methodcustom_forward
(*inputs)
utils/unet_blocks.py:1428
Methodcustom_forward
(*inputs)
utils/unet_blocks.py:1533
Functionddim_inversion
(pipeline, ddim_scheduler, video_latent, num_inv_steps, prompt="")
utils/util.py:80
Methoddisable_vae_slicing
(self)
utils/pipeline_magictime.py:114
Methodenable_sequential_cpu_offload
(self, gpu_id=0)
utils/pipeline_magictime.py:117
Methodenable_vae_slicing
(self)
utils/pipeline_magictime.py:111
Methodextra_repr
(self)
utils/unet_blocks.py:735
Functionextract_frame_number
(filename)
data_preprocess/step2_1_GPT4V_frame_caption.py:151
Methodfn_recursive_retrieve_slicable_dims
(module: torch.nn.Module)
utils/unet.py:265
Methodfn_recursive_set_attention_slice
(module: torch.nn.Module, slice_size: List[int])
utils/unet.py:303
Methodforward
r""" Args: sample (`torch.FloatTensor`): (batch, channel, height, width) noisy inputs tensor timestep (`torch.FloatTen
utils/unet.py:318
Methodforward
(self, hidden_states, encoder_hidden_states=None, timestep=None, return_dict: bool = True)
utils/unet_blocks.py:94
Methodforward
(self, hidden_states, encoder_hidden_states=None, timestep=None, attention_mask=None, video_length=None)
utils/unet_blocks.py:244
Methodforward
(self, x)
utils/unet_blocks.py:284
Methodforward
(self, x)
utils/unet_blocks.py:295
Methodforward
(self, hidden_states, output_size=None)
utils/unet_blocks.py:320
Methodforward
(self, hidden_states)
utils/unet_blocks.py:371
Methodforward
(self, input_tensor, temb)
utils/unet_blocks.py:455
Methodforward
(self, input_tensor, temb, encoder_hidden_states, attention_mask=None, anchor_frame_idx=None)
utils/unet_blocks.py:546
Methodforward
(self, hidden_states, encoder_hidden_states=None, attention_mask=None)
utils/unet_blocks.py:602
Methodforward
(self, hidden_states, encoder_hidden_states=None, attention_mask=None, video_length=None)
utils/unet_blocks.py:678
Methodforward
(self, x)
utils/unet_blocks.py:709
Methodforward
(self, hidden_states, encoder_hidden_states=None, attention_mask=None, video_length=None)
utils/unet_blocks.py:738
Methodforward
(self, hidden_states, temb=None, encoder_hidden_states=None, attention_mask=None)
utils/unet_blocks.py:1060
Methodforward
(self, hidden_states, temb=None, encoder_hidden_states=None, attention_mask=None)
utils/unet_blocks.py:1171
Methodforward
(self, hidden_states, temb=None, encoder_hidden_states=None)
utils/unet_blocks.py:1282
Methodforward
( self, hidden_states, res_hidden_states_tuple, temb=None, encoder_hid
utils/unet_blocks.py:1410
Methodforward
(self, hidden_states, res_hidden_states_tuple, temb=None, upsample_size=None, encoder_hidden_states=None,)
utils/unet_blocks.py:1524
Methodmagictime
( self, dreambooth_dropdown, motion_module_dropdown, prompt_textbox, n
app.py:173
Functionmerge_json_files
(info_data, caption_data)
data_preprocess/step2_2_preprocess_frame_caption.py:80
Functionmerge_json_files
(info_data, caption_data)
data_preprocess/step3_2_preprocess_video_caption.py:64
Functionsave_output
(video_id, prompt, output_file, api_key)
data_preprocess/step3_1_GPT4V_video_caption_concise.py:76
Functionsave_output
(video_id, prompt, output_file, api_key)
data_preprocess/step3_1_GPT4V_video_caption_detail.py:78
Functionsave_output
(video_id, prompt, output_file, api_key)
data_preprocess/step2_1_GPT4V_frame_caption.py:183
Methodset_use_memory_efficient_attention_xformers
(self, use_memory_efficient_attention_xformers: bool)
utils/unet_blocks.py:217
Functionzero_rank_print
(s)
utils/util.py:13
← previous101–144 of 144, ranked by callers