MCPcopy Create free account

hub / github.com/chaojie/ComfyUI-Open-Sora-Plan / functions

Functions1,253 in github.com/chaojie/ComfyUI-Open-Sora-Plan

↓ 3 callersMethod_predict_eps_from_xstart
(self, x_t, t, pred_xstart)
opensora/models/diffusion/diffusion/gaussian_diffusion.py:345
↓ 3 callersMethod_validate_input_list
Validates that the input is a list of expected type.
opensora/models/text_encoder/clip.py:78
↓ 3 callersMethod_vb_terms_bpd
Get a term for the variational lower-bound. The resulting units are bits (rather than nats, as one might expect). This allows
opensora/models/diffusion/diffusion/gaussian_diffusion_t2v.py:696
↓ 3 callersMethod_vb_terms_bpd
Get a term for the variational lower-bound. The resulting units are bits (rather than nats, as one might expect). This allows
opensora/models/diffusion/diffusion/gaussian_diffusion.py:686
↓ 3 callersFunctionbgr2ycbcr
Convert a BGR image to YCbCr image. The bgr version of rgb2ycbcr. It implements the ITU-R BT.601 conversion for standard-definition telev
opensora/models/super_resolution/basicsr/utils/matlab_functions.py:214
↓ 3 callersFunctioncalculate_fvd
(videos1, videos2, device, method='styleganv')
opensora/eval/cal_fvd.py:15
↓ 3 callersFunctioncast_tuple
(t, length = 1)
opensora/models/ae/videobase/causal_vqvae/modeling_causalvqvae.py:82
↓ 3 callersMethodcompute_pad
(self, dim, s)
opensora/eval/fvd/videogpt/pytorch_i3d.py:9
↓ 3 callersMethodcompute_pad
(self, dim, s)
opensora/eval/fvd/videogpt/pytorch_i3d.py:71
↓ 3 callersFunctionget_1d_sincos_pos_embed_from_grid
embed_dim: output dimension for each position pos: a list of positions to be encoded: size (M,) out: (M, D)
opensora/models/diffusion/utils/pos_embed.py:55
↓ 3 callersMethodget_input
(self, batch, k)
opensora/models/ae/videobase/causal_vae/modeling_causalvae.py:404
↓ 3 callersMethodim2win
(self, x, H, W)
opensora/models/super_resolution/basicsr/archs/rgt_arch.py:172
↓ 3 callersFunctionimfrombytes
Read an image from bytes. Args: content (bytes): Image bytes got from files or other streams. flag (str): Flags specifying the co
opensora/models/super_resolution/basicsr/utils/img_util.py:114
↓ 3 callersMethodkl
(self, other=None)
opensora/models/ae/videobase/utils/distrib_utils.py:19
↓ 3 callersMethodmake_layout
(self)
opensora/models/ae/videobase/causal_vqvae/modeling_causalvqvae.py:626
↓ 3 callersMethodmake_layout
(self)
opensora/models/ae/videobase/vqvae/modeling_vqvae.py:526
↓ 3 callersFunctionmean_flat
Take the mean over all non-batch dimensions.
opensora/models/diffusion/transport/utils.py:12
↓ 3 callersFunctionpad_to_multiple
(number, ds_stride)
opensora/utils/dataset_utils.py:37
↓ 3 callersMethodpreload
(self)
opensora/models/super_resolution/basicsr/data/prefetch_dataloader.py:105
↓ 3 callersFunctionreadPFM
(file)
opensora/models/frame_interpolation/utils/utils.py:120
↓ 3 callersMethodreset
(self)
opensora/models/frame_interpolation/utils/utils.py:16
↓ 3 callersMethodsample
Sampling x0 & t based on shape of x1 (if needed) Args: x1 - data point; [batch, *dim]
opensora/models/diffusion/transport/transport.py:103
↓ 3 callersFunctionscaled_dot_product_attention
(q, k, v, mask=None, attn_dropout=0.0, training=True)
opensora/models/ae/videobase/causal_vqvae/modeling_causalvqvae.py:59
↓ 3 callersFunctionscaled_dot_product_attention
(q, k, v, mask=None, attn_dropout=0.0, training=True)
opensora/models/ae/videobase/vqvae/modeling_vqvae.py:59
↓ 3 callersFunctionset_env
(seed=0)
opensora/serve/gradio_utils.py:6
↓ 3 callersMethodset_processor
r""" Set the attention processor to use. Args: processor (`AttnProcessor`): The attention processor to us
opensora/models/diffusion/latte/modules.py:517
↓ 3 callersFunctionshift_dim
(x, src_dim=-1, dest_dim=-1, make_contiguous=True)
opensora/models/ae/videobase/modules/ops.py:20
↓ 3 callersFunctionssim
(img1, img2)
opensora/eval/cal_ssim.py:6
↓ 3 callersFunctiontensor2im
(image_tensor, imtype=np.uint8, cent=1., factor=255./2.)
opensora/eval/flolpips/utils.py:17
↓ 3 callersMethodto_dict
(self)
opensora/models/ae/videobase/configuration_videobase.py:14
↓ 3 callersFunctiontrans
(x)
opensora/eval/cal_fvd.py:5
↓ 2 callersMethod__init__
Initializes Unit3D module.
opensora/eval/fvd/videogpt/pytorch_i3d.py:39
↓ 2 callersMethod__init__
(self, model, timestep_map, original_num_steps)
opensora/models/diffusion/diffusion/respace.py:120
↓ 2 callersMethod__init__
( self, z_channels: int, hidden_size: int, hidden_size_mult: Tuple[int] = (1,
opensora/models/ae/videobase/causal_vae/modeling_causalvae.py:14
↓ 2 callersMethod__init__
( self, logvar_init=0.0, kl_weight=1.0, pixelloss_weight=1.0, perceptu
opensora/models/ae/videobase/losses/perceptual_loss.py:363
↓ 2 callersMethod__init__
(self, loader)
opensora/models/super_resolution/basicsr/data/prefetch_dataloader.py:70
↓ 2 callersMethod__init__
(self, fmap1, fmap2, num_levels=4, radius=4)
opensora/models/frame_interpolation/networks/blocks/raft.py:143
↓ 2 callersMethod_clean_caption
(self, caption)
opensora/sample/pipeline_videogen.py:386
↓ 2 callersMethod_combine_without_prefix
(self, folder_path, prefix='.')
opensora/eval/eval_common_metric.py:116
↓ 2 callersMethod_combine_without_prefix
(self, folder_path, prefix='.')
opensora/eval/eval_clip_score.py:119
↓ 2 callersMethod_do_register
(self, name, obj)
opensora/models/super_resolution/basicsr/utils/registry.py:38
↓ 2 callersMethod_load_modality
(self, path, modality)
opensora/eval/eval_clip_score.py:88
↓ 2 callersMethod_load_video
(self, video_path)
opensora/eval/eval_common_metric.py:90
↓ 2 callersMethod_predict_xstart_from_eps
(self, x_t, t, eps)
opensora/models/diffusion/diffusion/gaussian_diffusion_t2v.py:348
↓ 2 callersMethod_predict_xstart_from_eps
(self, x_t, t, eps)
opensora/models/diffusion/diffusion/gaussian_diffusion.py:338
↓ 2 callersFunction_symmetric_matrix_square_root
(mat, eps=1e-10)
opensora/eval/fvd/videogpt/fvd.py:68
↓ 2 callersMethod_text_preprocessing
(self, text, clean_caption=False)
opensora/sample/pipeline_videogen.py:361
↓ 2 callersMethod_tile
(self, x)
opensora/models/ae/videobase/causal_vqvae/modeling_causalvqvae.py:266
↓ 2 callersMethod_tile
(self, x)
opensora/models/ae/videobase/modules/quant.py:19
↓ 2 callersMethod_tile
(self, x)
opensora/models/ae/videobase/vqvae/modeling_vqvae.py:132
↓ 2 callersMethod_to_flattened_idx
(self, idx)
opensora/models/ae/videobase/causal_vqvae/modeling_causalvqvae.py:681
↓ 2 callersMethod_to_flattened_idx
(self, idx)
opensora/models/ae/videobase/vqvae/modeling_vqvae.py:581
↓ 2 callersMethod_to_unflattened_idx
(self, flat_idx)
opensora/models/ae/videobase/causal_vqvae/modeling_causalvqvae.py:690
↓ 2 callersMethod_to_unflattened_idx
(self, flat_idx)
opensora/models/ae/videobase/vqvae/modeling_vqvae.py:590
↓ 2 callersFunction_totensor
(img, bgr2rgb, float32)
opensora/models/super_resolution/basicsr/utils/img_util.py:22
↓ 2 callersMethod_unpad
(self, x)
opensora/models/frame_interpolation/utils/utils.py:74
↓ 2 callersFunction_warmup_beta
(beta_start, beta_end, num_diffusion_timesteps, warmup_frac)
opensora/models/diffusion/diffusion/gaussian_diffusion_t2v.py:58
↓ 2 callersFunction_warmup_beta
(beta_start, beta_end, num_diffusion_timesteps, warmup_frac)
opensora/models/diffusion/diffusion/gaussian_diffusion.py:58
↓ 2 callersMethodapply_condition
(self, size: torch.Tensor, batch_size: int, embedder: nn.Module)
opensora/models/diffusion/latte/modules.py:54
↓ 2 callersMethodapply_rope1d
(self, tokens, pos1d, cos, sin)
opensora/models/diffusion/utils/pos_embed.py:113
↓ 2 callersFunctionapprox_standard_normal_cdf
A fast approximation of the cumulative distribution function of the standard normal.
opensora/models/diffusion/diffusion/diffusion_utils.py:39
↓ 2 callersFunctionarray_to_video
(image_array: npt.NDArray, fps: float = 30.0, output_file: str = 'output_video.mp4')
examples/rec_imvi_vae.py:22
↓ 2 callersFunctionbicubic_upsample_opencv
(input_image_path, output_image_path, scale_factor)
opensora/models/super_resolution/run.py:38
↓ 2 callersFunctionbilinear_sampler
Wrapper for grid_sample, uses pixel coordinates
opensora/models/frame_interpolation/networks/blocks/raft.py:10
↓ 2 callersMethodblend_h
( self, a: torch.Tensor, b: torch.Tensor, blend_extent: int )
opensora/models/ae/videobase/causal_vae/modeling_causalvae.py:548
↓ 2 callersMethodblend_v
( self, a: torch.Tensor, b: torch.Tensor, blend_extent: int )
opensora/models/ae/videobase/causal_vae/modeling_causalvae.py:538
↓ 2 callersFunctionbuild_loss
Build loss from options. Args: opt (dict): Configuration. It must contain: type (str): Model type.
opensora/models/super_resolution/basicsr/losses/__init__.py:14
↓ 2 callersFunctionbuild_network
(opt)
opensora/models/super_resolution/basicsr/archs/__init__.py:19
↓ 2 callersFunctioncalculate_flolpips
(videos1, videos2, device)
opensora/eval/cal_flolpips.py:18
↓ 2 callersFunctioncalculate_lpips
(videos1, videos2, device)
opensora/eval/cal_lpips.py:25
↓ 2 callersMethodcalculate_mask
(self, H, W)
opensora/models/super_resolution/basicsr/archs/rgt_arch.py:269
↓ 2 callersFunctioncalculate_psnr
(videos1, videos2)
opensora/eval/cal_psnr.py:20
↓ 2 callersFunctioncalculate_ssim
(videos1, videos2)
opensora/eval/cal_ssim.py:47
↓ 2 callersFunctioncalculate_weights_indices
Calculate weights and indices, used for imresize function. Args: in_length (int): Input length. out_length (int): Output length.
opensora/models/super_resolution/basicsr/utils/matlab_functions.py:16
↓ 2 callersFunctioncaption_summary
apply GPT3-Turbo as the combination for original caption and the prompted captions for a video
opensora/models/captioner/caption_refiner/gpt_combinator.py:52
↓ 2 callersFunctioncenter_crop
(clip, crop_size)
opensora/dataset/transform.py:82
↓ 2 callersFunctionclean_caption
(caption)
opensora/utils/utils.py:371
↓ 2 callersMethodclean_caption
(self, caption)
opensora/models/text_encoder/t5.py:79
↓ 2 callersFunctioncompute_stats
(feats: np.ndarray)
opensora/eval/fvd/styleganv/fvd.py:75
↓ 2 callersMethodcondition_score
Compute what the p_mean_variance output would have been, should the model's score function be conditioned by cond_fn. See con
opensora/models/diffusion/diffusion/gaussian_diffusion_t2v.py:372
↓ 2 callersMethodcondition_score
Compute what the p_mean_variance output would have been, should the model's score function be conditioned by cond_fn. See con
opensora/models/diffusion/diffusion/gaussian_diffusion.py:362
↓ 2 callersFunctioncov
Estimate a covariance matrix given data. Covariance indicates the level to which two variables vary together. If we examine N-dimensional sam
opensora/eval/fvd/videogpt/fvd.py:80
↓ 2 callersFunctioncreate_temp_folder
(folder_path)
opensora/models/super_resolution/run.py:25
↓ 2 callersFunctioncubic
cubic function used for calculate_weights_indices.
opensora/models/super_resolution/basicsr/utils/matlab_functions.py:6
↓ 2 callersFunctioncustom_to_video
( x: torch.Tensor, fps: float = 2.0, output_file: str = "output_video.mp4" )
examples/rec_video_vae.py:36
↓ 2 callersMethoddecode
(self, z)
opensora/models/ae/videobase/causal_vae/modeling_causalvae.py:384
↓ 2 callersFunctiondiscretized_gaussian_log_likelihood
Compute the log-likelihood of a Gaussian distribution discretizing to a given image. :param x: the target images. It is assumed that this
opensora/models/diffusion/diffusion/diffusion_utils.py:62
↓ 2 callersMethoddownload_and_load_model
(cls, model_name, cache_dir=None)
opensora/models/ae/videobase/modeling_videobase.py:30
↓ 2 callersFunctiondssim
(p0, p1, range=255.)
opensora/eval/flolpips/utils.py:13
↓ 2 callersMethodencode
(self, x)
opensora/models/ae/videobase/causal_vae/modeling_causalvae.py:372
↓ 2 callersMethodforward
input: * tokens: batch_size x nheads x ntokens x dim * positions: batch_size x ntokens x 2 (y and x posit
opensora/models/diffusion/utils/pos_embed.py:120
↓ 2 callersFunctionforward_modality
(model, data, flag)
opensora/eval/eval_clip_score.py:154
↓ 2 callersFunctionget_2d_sincos_pos_embed
grid_size: int of the grid height and width return: pos_embed: [grid_size*grid_size, embed_dim] or [1+grid_size*grid_size, embed_dim] (w/ or
opensora/models/diffusion/utils/pos_embed.py:14
↓ 2 callersFunctionget_ckpt_path
(name, root, check=False)
opensora/utils/taming_download.py:38
↓ 2 callersFunctionget_fvd_feats
(videos, i3d, device, bs=10)
opensora/eval/fvd/styleganv/fvd.py:31
↓ 2 callersMethodget_image
(self, idx)
opensora/dataset/t2v_datasets.py:106
↓ 2 callersMethodget_img_cap_list
(self)
opensora/dataset/t2v_datasets.py:189
↓ 2 callersMethodget_last_layer
(self)
opensora/models/ae/videobase/causal_vae/modeling_causalvae.py:532
↓ 2 callersMethodget_ops
(self)
opensora/models/ae/videobase/causal_vqvae/modeling_causalvqvae.py:728
↓ 2 callersMethodget_ops
(self)
opensora/models/ae/videobase/vqvae/modeling_vqvae.py:629
← previousnext →101–200 of 1,253, ranked by callers