MCPcopy Create free account

hub / github.com/Doubiiu/ToonCrafter / functions

Functions475 in github.com/Doubiiu/ToonCrafter

⨍ Functions 475◇ Types & classes 111
↓ 1 callersMethod_load_metadata
(self)
lvdm/data/webvid.py:72
↓ 1 callersMethod_make_conv
(self)
lvdm/models/autoencoder_dualref.py:483
↓ 1 callersMethodattention
(self, h_: torch.Tensor)
lvdm/models/autoencoder_dualref.py:172
↓ 1 callersFunctionavg_pool_nd
Create a 1D, 2D, or 3D average pooling module.
lvdm/basics.py:56
↓ 1 callersMethodconfigure_schedulers
(self, optimizer)
lvdm/models/ddpm3d.py:1015
↓ 1 callersMethodcopy_to
(self, model)
lvdm/ema.py:46
↓ 1 callersMethodddim_sampling
(self, cond, shape, x_T=None, ddim_use_original_steps=False, cal
lvdm/models/samplers/ddim_multiplecond.py:138
↓ 1 callersMethodddim_sampling
(self, cond, shape, x_T=None, ddim_use_original_steps=False, callb
lvdm/models/samplers/ddim.py:135
↓ 1 callersMethoddownload_model
(self)
scripts/gradio/i2v_test.py:94
↓ 1 callersMethoddownload_model
(self)
scripts/gradio/i2v_test_application.py:117
↓ 1 callersFunctiondynamicrafter_demo
(result_dir='./tmp/', res=512)
gradio_app.py:17
↓ 1 callersMethodencode_with_pretrained
(self,x)
lvdm/modules/networks/ae_modules.py:838
↓ 1 callersMethodencode_with_transformer
(self, text)
lvdm/modules/encoders/condition.py:214
↓ 1 callersMethodencode_with_vision_transformer
(self, img)
lvdm/modules/encoders/condition.py:287
↓ 1 callersMethodencode_with_vision_transformer
(self, x)
lvdm/modules/encoders/condition.py:342
↓ 1 callersFunctionequals
(val)
lvdm/modules/x_transformer.py:75
↓ 1 callersMethodfreeze
(self)
lvdm/modules/encoders/condition.py:70
↓ 1 callersMethodfreeze
(self)
lvdm/modules/encoders/condition.py:113
↓ 1 callersMethodfreeze
(self)
lvdm/modules/encoders/condition.py:204
↓ 1 callersMethodfreeze
(self)
lvdm/modules/encoders/condition.py:275
↓ 1 callersMethodfreeze
(self)
lvdm/modules/encoders/condition.py:332
↓ 1 callersMethodget_alpha
(self, bs)
lvdm/models/autoencoder_dualref.py:884
↓ 1 callersMethodget_alpha
( self, )
lvdm/models/autoencoder_dualref.py:997
↓ 1 callersMethodget_alpha
( self, )
lvdm/models/autoencoder_dualref.py:1067
↓ 1 callersFunctionget_dist_info
()
scripts/evaluation/ddp_wrapper.py:15
↓ 1 callersMethodget_image
(self, image, prompt, steps=50, cfg_scale=7.5, eta=1.0, fs=3, seed=123)
scripts/gradio/i2v_test.py:37
↓ 1 callersMethodget_image
(self, image, prompt, steps=50, cfg_scale=7.5, eta=1.0, fs=3, seed=123, image2=None)
scripts/gradio/i2v_test_application.py:38
↓ 1 callersFunctionget_latent_z
(model, videos)
scripts/evaluation/funcs.py:234
↓ 1 callersFunctionget_latent_z_with_hidden_states
(model, videos)
scripts/evaluation/inference.py:164
↓ 1 callersMethodget_latent_z_with_hidden_states
(self, model, videos)
scripts/gradio/i2v_test_application.py:127
↓ 1 callersFunctionget_nondefault_trainer_args
(args)
main/trainer.py:33
↓ 1 callersFunctionget_obj_from_str
(string, reload=False)
utils/utils.py:37
↓ 1 callersFunctionget_parser
()
gradio_app.py:70
↓ 1 callersFunctionget_parser
(**parser_kwargs)
main/trainer.py:14
↓ 1 callersFunctionget_parser
()
scripts/evaluation/inference.py:347
↓ 1 callersFunctionget_timestep_embedding
This matches the implementation in Denoising Diffusion Probabilistic Models: From Fairseq. Build sinusoidal embeddings. This matches
lvdm/modules/networks/ae_modules.py:131
↓ 1 callersFunctionget_trainer_callbacks
(lightning_config, config, logdir, ckptdir, logger)
main/utils_train.py:35
↓ 1 callersFunctionget_trainer_logger
(lightning_config, logdir, on_debug)
main/utils_train.py:99
↓ 1 callersFunctionget_trainer_strategy
(lightning_config)
main/utils_train.py:125
↓ 1 callersFunctionimage_guided_synthesis
(model, prompts, videos, noise_shape, n_samples=1, ddim_steps=50, ddim_eta=1., \ unco
scripts/evaluation/inference.py:180
↓ 1 callersMethodinit_
(self)
lvdm/modules/x_transformer.py:30
↓ 1 callersMethodinit_
(self)
lvdm/modules/x_transformer.py:594
↓ 1 callersMethodinit_from_ckpt
(self, path, ignore_keys=list())
lvdm/models/autoencoder.py:83
↓ 1 callersMethodinit_test
(self,)
lvdm/models/autoencoder.py:54
↓ 1 callersFunctioninit_workspace
(name, logdir, model_config, lightning_config, rank=0)
main/utils_train.py:9
↓ 1 callersMethodinstantiate_cond_stage
(self, config)
lvdm/models/ddpm3d.py:588
↓ 1 callersMethodinstantiate_first_stage
(self, config)
lvdm/models/ddpm3d.py:581
↓ 1 callersMethodinstantiate_pretrained
(self, config)
lvdm/modules/networks/ae_modules.py:829
↓ 1 callersFunctionload_checkpoint
(model, ckpt, full_strict)
scripts/evaluation/funcs.py:114
↓ 1 callersFunctionload_checkpoints
(model, model_cfg)
main/utils_train.py:138
↓ 1 callersFunctionload_data_prompts
(data_dir, video_size=(256,256), video_frames=16, interp=False)
scripts/evaluation/inference.py:64
↓ 1 callersFunctionload_model_checkpoint
(model, ckpt)
scripts/evaluation/inference.py:27
↓ 1 callersFunctionload_num_videos
(data_path, num_videos)
utils/save_video.py:150
↓ 1 callersFunctionload_prompts
(prompt_file)
scripts/evaluation/inference.py:54
↓ 1 callersFunctionlog_local
(batch_logs, save_dir, filename, save_fps=10, rescale=True)
utils/save_video.py:62
↓ 1 callersMethodlog_to_tensorboard
log images and videos to tensorboard
main/callbacks.py:31
↓ 1 callersFunctionmake_beta_schedule
(schedule, n_timestep, linear_start=1e-4, linear_end=2e-2, cosine_s=8e-3)
lvdm/models/utils_diffusion.py:31
↓ 1 callersMethodmake_cond_schedule
(self, )
lvdm/models/ddpm3d.py:549
↓ 1 callersMethodmake_schedule
(self, ddim_num_steps, ddim_discretize="uniform", ddim_eta=0., verbose=True)
lvdm/models/samplers/ddim_multiplecond.py:24
↓ 1 callersMethodmake_schedule
(self, ddim_num_steps, ddim_discretize="uniform", ddim_eta=0., verbose=True)
lvdm/models/samplers/ddim.py:24
↓ 1 callersFunctionmax_neg_value
(tensor)
lvdm/modules/x_transformer.py:81
↓ 1 callersFunctionnot_equals
(val)
lvdm/modules/x_transformer.py:69
↓ 1 callersMethodp_losses
(self, x_start, t, noise=None)
lvdm/models/ddpm3d.py:332
↓ 1 callersMethodp_losses
(self, x_start, cond, t, noise=None, **kwargs)
lvdm/models/ddpm3d.py:752
↓ 1 callersMethodp_mean_variance
(self, x, t, clip_denoised: bool)
lvdm/models/ddpm3d.py:263
↓ 1 callersMethodp_mean_variance
(self, x, c, t, clip_denoised: bool, return_x0=False, score_corrector=None, corrector_kwargs=None, **kwargs)
lvdm/models/ddpm3d.py:892
↓ 1 callersMethodp_sample
(self, x, t, clip_denoised=True, repeat_noise=False)
lvdm/models/ddpm3d.py:276
↓ 1 callersMethodp_sample
(self, x, c, t, clip_denoised=False, repeat_noise=False, return_x0=False, \ temperature=1., n
lvdm/models/ddpm3d.py:918
↓ 1 callersMethodp_sample_loop
(self, shape, return_intermediates=False)
lvdm/models/ddpm3d.py:285
↓ 1 callersMethodp_sample_loop
(self, cond, shape, return_intermediates=False, x_T=None, verbose=True, callback=None, \
lvdm/models/ddpm3d.py:940
↓ 1 callersFunctionprepare_to_log
(batch_logs, max_images=100000, clamp=True)
utils/save_video.py:120
↓ 1 callersMethodpreprocess
(self, x)
lvdm/modules/encoders/condition.py:155
↓ 1 callersMethodpreprocess
(self, x)
lvdm/modules/encoders/condition.py:265
↓ 1 callersMethodpreprocess
(self, x)
lvdm/modules/encoders/condition.py:322
↓ 1 callersFunctionread_first_n_frames
(d: os.PathLike, num_frames: int)
utils/save_video.py:15
↓ 1 callersFunctionrescale_zero_terminal_snr
Rescales betas to have zero terminal SNR Based on https://arxiv.org/pdf/2305.08891.pdf (Algorithm 1) Args: betas (`numpy.ndarray`):
lvdm/models/utils_diffusion.py:112
↓ 1 callersMethodrestore
Restore the parameters stored with the `store` method. Useful to validate the model with EMA parameters without affecting the
lvdm/ema.py:64
↓ 1 callersFunctionrun_inference
(args, gpu_num, gpu_no)
scripts/evaluation/inference.py:280
↓ 1 callersMethodsample
(self, cond, batch_size=16, return_intermediates=False, x_T=None, \ verbose=True, timesteps=Non
lvdm/models/ddpm3d.py:987
↓ 1 callersFunctionsave_img_grid
(grid, path, rescale)
utils/save_video.py:66
↓ 1 callersFunctionsave_results_seperate
(prompt, samples, filename, fakedir, fps=10, loop=False)
scripts/evaluation/inference.py:135
↓ 1 callersFunctionset_logger
(logfile, name='mainlogger')
main/utils_train.py:162
↓ 1 callersMethodsetup
(self, stage=None)
main/utils_data.py:75
↓ 1 callersFunctionsetup_dist
(local_rank)
scripts/evaluation/ddp_wrapper.py:8
↓ 1 callersMethodshared_step
(self, batch, random_uncond, **kwargs)
lvdm/models/ddpm3d.py:729
↓ 1 callersMethodstore
Save the current parameters for restoring later. Args: parameters: Iterable of `torch.nn.Parameter`; the parameters to be
lvdm/ema.py:55
↓ 1 callersFunctiontensor_to_mp4
video: torch.Tensor, b,c,t,h,w, 0-1 if -1~1, enable rescale=True
utils/save_video.py:27
↓ 1 callersMethodtext_transformer_forward
(self, x: torch.Tensor, attn_mask=None)
lvdm/modules/encoders/condition.py:223
Method__getitem__
(self, idx)
main/utils_data.py:40
Method__getitem__
(self, index)
lvdm/data/webvid.py:88
Method__init__
(self, batch_size, train=None, validation=None, test=None, predict=None, wrap=False, num_wor
main/utils_data.py:45
Method__init__
(self, batch_frequency, max_images=8, clamp=True, rescale=True, save_dir=None, \ to_local=Fal
main/callbacks.py:16
Method__init__
(self,result_dir='./tmp/',gpu_num=1,resolution='256_256')
scripts/gradio/i2v_test.py:14
Method__init__
(self,result_dir='./tmp/',gpu_num=1,resolution='256_256')
scripts/gradio/i2v_test_application.py:14
Method__init__
(self, model, decay=0.9999, use_num_upates=True)
lvdm/ema.py:6
Method__init__
(self, c_concat_config, c_crossattn_config)
lvdm/basics.py:92
Method__init__
(self, value)
lvdm/distributions.py:14
Method__init__
(self, parameters, deterministic=False)
lvdm/distributions.py:25
Method__init__
(self, dim, max_seq_len)
lvdm/modules/x_transformer.py:25
Method__init__
(self, dim)
lvdm/modules/x_transformer.py:39
← previousnext →101–200 of 475, ranked by callers