MCPcopy Create free account

hub / github.com/WuTao-CS/CustomCrafter / functions

Functions992 in github.com/WuTao-CS/CustomCrafter

⨍ Functions 992◇ Types & classes 193
↓ 1 callersMethodreplace_cross_attention
(self, attn_base, att_replace)
lvdm/utils/ptp_utils.py:340
↓ 1 callersMethodreplace_self_attention
(self, attn_base, att_replace)
lvdm/utils/ptp_utils.py:259
↓ 1 callersMethodreset_parameters
(self)
extralibs/model_edge.py:373
↓ 1 callersMethodresize_sparse_flow_map
(self, flow, valid, fx=1.0, fy=1.0)
eval/RAFT/core/utils_core/augmentor.py:161
↓ 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
(args, run_id, i3d, device)
utils/cal_fvd_text2video.py:42
↓ 1 callersFunctionrun_inference
(args, gpu_num, gpu_no)
pipeline/evaluation/timestep_inference_lora.py:228
↓ 1 callersFunctionrun_inference
(args, gpu_num, gpu_no)
pipeline/evaluation/inference_inter.py:183
↓ 1 callersFunctionrun_inference
(args, gpu_num, gpu_no)
pipeline/evaluation/timestep_inference.py:211
↓ 1 callersFunctionrun_inference
(args, gpu_num, gpu_no, prompt_list=['a <new1> man is riding a bike, heading towards the camera. medium shot',
pipeline/evaluation/p2p_inference.py:190
↓ 1 callersFunctionrun_inference
(args, gpu_num, gpu_no)
pipeline/evaluation/base_inference_inter.py:181
↓ 1 callersFunctionrun_inference
(args, gpu_num, gpu_no, prompt_list=['a new teddybear is playing guitar. Complete picture','a <new1> teddybear
pipeline/evaluation/p2p_inference_timesteps.py:224
↓ 1 callersFunctionrun_inference
(args, gpu_num, gpu_no)
pipeline/evaluation/inference.py:166
↓ 1 callersFunctionrun_inference
(args, gpu_num, gpu_no)
pipeline/evaluation/image_inference.py:166
↓ 1 callersFunctionrun_inference
(args, gpu_num, gpu_no)
pipeline/evaluation/base_inference.py:176
↓ 1 callersFunctionrun_multitimes
(args, i3d, device)
utils/cal_fvd_text2video.py:139
↓ 1 callersMethodsample
(self, cond, batch_size=16, return_intermediates=False, x_T=None, \ verbose=True, timesteps=Non
lvdm/models/ddpm3d.py:1121
↓ 1 callersMethodsample
Compute the sample at time `t_end` by DPM-Solver, given the initial `x` at time `t_start`. =========================================
lvdm/models/samplers/dpm_solver/dpm_solver.py:992
↓ 1 callersMethodsample_log
(self, cond, batch_size, ddim, ddim_steps, **kwargs)
lvdm/models/ddpm3d.py:1138
↓ 1 callersFunctionsave_img_grid
(grid, path, rescale)
utils/save_video.py:122
↓ 1 callersFunctionsave_results
(prompt, samples, inputs, filename, realdir, fakedir, fps=10)
pipeline/evaluation/p2p_inference.py:82
↓ 1 callersFunctionsave_results
(prompt, samples, inputs, filename, realdir, fakedir, fps=10)
pipeline/evaluation/p2p_inference_timesteps.py:82
↓ 1 callersMethodset_params
(self, frame, count)
eval/eval_dd.py:50
↓ 1 callersMethodset_selector_from_diag
(self, diag: torch.Tensor)
lvdm/models/lora.py:62
↓ 1 callersMethodsetup
(self, stage=None)
custom.py:392
↓ 1 callersFunctionsetup_dist
(local_rank)
pipeline/evaluation/ddp_wrapper.py:10
↓ 1 callersMethodshared_step
(self, batch)
lvdm/models/ddpm3d.py:376
↓ 1 callersMethodshared_step
(self, batch, random_uncond, **kwargs)
lvdm/models/ddpm3d.py:800
↓ 1 callersMethodsinglestep_dpm_solver_update
Singlestep DPM-Solver with the order `order` from time `s` to time `t`. Args: x: A pytorch tensor. The initial value at
lvdm/models/samplers/dpm_solver/dpm_solver.py:876
↓ 1 callersMethodspatial_transform
(self, img1, img2, flow)
eval/RAFT/core/utils_core/augmentor.py:67
↓ 1 callersMethodspatial_transform
(self, img1, img2, flow, valid)
eval/RAFT/core/utils_core/augmentor.py:195
↓ 1 callersMethodstate_dict
(self, *args, **kwargs)
lvdm/modules/embedding_manager.py:149
↓ 1 callersMethodstore
Save the current parameters for restoring later. Args: parameters: Iterable of `torch.nn.Parameter`; the parameters to b
lvdm/ema.py:55
↓ 1 callersFunctionsynset2idx
(path_to_yaml="data/index_synset.yaml")
lvdm/data/imagenet.py:20
↓ 1 callersFunctiontest3
()
extralibs/model_edge.py:754
↓ 1 callersMethodtext_transformer_forward
(self, x: torch.Tensor, attn_mask=None)
lvdm/modules/encoders/condition.py:374
↓ 1 callersMethodtext_transformer_forward
(self, x: torch.Tensor, attn_mask=None)
lvdm/modules/encoders/custom_condition.py:225
↓ 1 callersMethodtext_transformer_forward
(self, x: torch.Tensor, attn_mask=None)
lvdm/modules/encoders/custom_condition.py:398
↓ 1 callersMethodtext_transformer_forward
(self, x: torch.Tensor, attn_mask=None)
lvdm/modules/encoders/custom_condition.py:559
↓ 1 callersFunctiontrace_sqrt_product
(sigma, sigma_v)
utils/fvd_utils.py:73
↓ 1 callersFunctionupflow8
(flow, mode='bilinear')
eval/RAFT/core/utils_core/utils.py:80
↓ 1 callersMethodupsample_flow
Upsample flow field [H/8, W/8, 2] -> [H, W, 2] using convex combination
eval/RAFT/core/raft.py:72
↓ 1 callersFunctionwrite_pfm
Write pfm file. Args: path (str): pathto file image (array): data scale (int, optional): Scale. Defaults to 1.
extralibs/midas/utils.py:58
FunctionPYBIND11_MODULE
eval/RAFT/alt_cuda_corr/correlation.cpp:51
Method__call__
(self, coords)
eval/RAFT/core/corr.py:29
Method__call__
(self, coords)
eval/RAFT/core/corr.py:74
Method__call__
(self, img1, img2, flow)
eval/RAFT/core/utils_core/augmentor.py:111
Method__call__
(self, img1, img2, flow, valid)
eval/RAFT/core/utils_core/augmentor.py:236
Method__call__
(self, *args)
lvdm/utils/ptp_utils.py:109
Method__call__
(self, attn, is_cross: bool, place_in_unet: str)
lvdm/utils/ptp_utils.py:158
Method__call__
(self, x_t, attention_store)
lvdm/utils/ptp_utils.py:223
Method__call__
(self, sample)
lvdm/data/util.py:19
Method__call__
(self, sample)
extralibs/midas/midas/transforms.py:162
Method__call__
(self, sample)
extralibs/midas/midas/transforms.py:205
Method__call__
(self, sample)
extralibs/midas/midas/transforms.py:218
Method__delattr__
(self, name: str)
eval/dnnlib/util.py:52
Method__enter__
(self)
eval/RAFT/core/raft.py:18
Method__enter__
(self)
eval/dnnlib/util.py:75
Method__exit__
(self, *args)
eval/RAFT/core/raft.py:20
Method__exit__
(self, exc_type: Any, exc_value: Any, traceback: Any)
eval/dnnlib/util.py:78
Method__getitem__
(self, idx)
custom.py:303
Method__getitem__
(self, idx)
custom.py:309
Method__getitem__
(self, index)
eval/RAFT/core/datasets.py:34
Method__getitem__
(self, i)
lvdm/data/imagenet.py:42
Method__getitem__
(self, i)
lvdm/data/imagenet.py:339
Method__getitem__
(self, i)
lvdm/data/pseudodata.py:25
Method__getitem__
(self, index)
lvdm/data/webvid_data.py:113
Method__getitem__
(self, i)
lvdm/data/lsun.py:39
Method__getitem__
(self, i)
lvdm/data/personalized.py:180
Method__getitem__
(self, index)
lvdm/data/customdata_new.py:194
Method__init__
(self, dataset)
custom.py:297
Method__init__
(self, batch_size, train=None, train2=None, validation=None, test=None, predict=None, wrap=Fa
custom.py:363
Method__init__
(self, resume, now, logdir, ckptdir, cfgdir, config, lightning_config)
custom.py:462
Method__init__
(self, device=None)
utils/eval_utils.py:38
Method__init__
(self, args, device)
eval/eval_dd.py:18
Method__init__
(self, device= "cuda" if torch.cuda.is_available() else "cpu" )
eval/eval_clip.py:28
Method__init__
(self, device="cuda" if torch.cuda.is_available() else "cpu", vit_model='checkpoints/dino-vits16')
eval/eval_dino.py:22
Method__init__
(self, in_planes, planes, norm_fn='group', stride=1)
eval/RAFT/core/extractor.py:7
Method__init__
(self, in_planes, planes, norm_fn='group', stride=1)
eval/RAFT/core/extractor.py:61
Method__init__
(self, output_dim=128, norm_fn='batch', dropout=0.0)
eval/RAFT/core/extractor.py:196
Method__init__
(self, aug_params=None, sparse=False)
eval/RAFT/core/datasets.py:19
Method__init__
(self, aug_params=None, split='training', root='datasets/Sintel', dstype='clean')
eval/RAFT/core/datasets.py:103
Method__init__
(self, aug_params=None, split='train', root='datasets/FlyingChairs_release/data')
eval/RAFT/core/datasets.py:122
Method__init__
(self, aug_params=None, root='datasets/FlyingThings3D', dstype='frames_cleanpass')
eval/RAFT/core/datasets.py:138
Method__init__
(self, aug_params=None, split='training', root='datasets/KITTI')
eval/RAFT/core/datasets.py:162
Method__init__
(self, fmap1, fmap2, num_levels=4, radius=4)
eval/RAFT/core/corr.py:13
Method__init__
(self, fmap1, fmap2, num_levels=4, radius=4)
eval/RAFT/core/corr.py:64
Method__init__
(self, input_dim=128, hidden_dim=256)
eval/RAFT/core/update.py:7
Method__init__
(self, hidden_dim=128, input_dim=192+128)
eval/RAFT/core/update.py:34
Method__init__
(self, args)
eval/RAFT/core/update.py:63
Method__init__
(self, args)
eval/RAFT/core/update.py:80
Method__init__
(self, args, hidden_dim=96)
eval/RAFT/core/update.py:100
Method__init__
(self, args, hidden_dim=128, input_dim=128)
eval/RAFT/core/update.py:115
Method__init__
(self, enabled)
eval/RAFT/core/raft.py:16
Method__init__
(self, args)
eval/RAFT/core/raft.py:25
Method__init__
(self, crop_size, min_scale=-0.2, max_scale=0.5, do_flip=True)
eval/RAFT/core/utils_core/augmentor.py:16
Method__init__
(self, crop_size, min_scale=-0.2, max_scale=0.5, do_flip=False)
eval/RAFT/core/utils_core/augmentor.py:123
Method__init__
(self, dims, mode='sintel')
eval/RAFT/core/utils_core/utils.py:9
Method__init__
(self, file_name: str = None, file_mode: str = "w", should_flush: bool = True)
eval/dnnlib/util.py:62
Method__init__
(self, model, decay=0.9999, use_num_upates=True)
lvdm/ema.py:6
← previousnext →401–500 of 992, ranked by callers