MCPcopy Create free account

hub / github.com/Vchitect/LaVie / functions

Functions553 in github.com/Vchitect/LaVie

↓ 1 callersMethodset_attention_slice
r""" Enable sliced attention computation. When this option is enabled, the attention module will split the input tensor in slices, to
interpolation/models/unet.py:244
↓ 1 callersFunctionspace_timesteps
Create a list of timesteps to use from an original diffusion process, given the number of timesteps we want to take from equally-sized portio
vsr/diffusion/respace.py:12
↓ 1 callersFunctionspace_timesteps
Create a list of timesteps to use from an original diffusion process, given the number of timesteps we want to take from equally-sized portio
interpolation/diffusion/respace.py:12
↓ 1 callersMethodtiled_decode
r"""Decode a batch of images using a tiled decoder. Args: When this option is enabled, the VAE will split the input tensor into tiles to compute
vsr/models/autoencoder_kl.py:261
↓ 1 callersMethodtiled_encode
r"""Encode a batch of images using a tiled encoder. Args: When this option is enabled, the VAE will split the input tensor into tiles to compute
vsr/models/autoencoder_kl.py:214
↓ 1 callersFunctionto_tensor
Convert tensor data type from uint8 to float, divide value by 255.0 and permute the dimensions of clip tensor Args: clip (torch.t
interpolation/datasets/video_transforms.py:16
↓ 1 callersMethodtoken_drop
Drops text to enable classifier-free guidance.
base/models/clip.py:70
↓ 1 callersMethodtoken_drop
Drops text to enable classifier-free guidance.
vsr/models/clip.py:73
↓ 1 callersMethodtoken_drop
Drops text to enable classifier-free guidance.
interpolation/models/clip.py:69
↓ 1 callersMethodupdate_with_all_losses
Update the reweighting using losses from a model. Sub-classes should override this method to update the reweighting using los
vsr/diffusion/timestep_sampler.py:106
↓ 1 callersMethodupdate_with_all_losses
Update the reweighting using losses from a model. Sub-classes should override this method to update the reweighting using los
interpolation/diffusion/timestep_sampler.py:106
↓ 1 callersMethodweights
Get a numpy array of weights, one per diffusion step. The weights needn't be normalized, but must be positive.
vsr/diffusion/timestep_sampler.py:38
↓ 1 callersMethodweights
Get a numpy array of weights, one per diffusion step. The weights needn't be normalized, but must be positive.
interpolation/diffusion/timestep_sampler.py:38
Method__call__
r""" Function invoked when calling the pipeline for generation. Args: prompt (`str` or `List[str]`, *optional*):
base/pipelines/pipeline_videogen.py:498
Method__call__
(self, x, ts, **kwargs)
vsr/diffusion/respace.py:125
Method__call__
r""" Function invoked when calling the pipeline for generation. Args: prompt (`str` or `List[str]`, *optional*):
vsr/models/pipeline_stable_diffusion_upscale_video_3d.py:492
Method__call__
(self, x, ts, **kwargs)
interpolation/diffusion/respace.py:125
Method__call__
Args: clip (torch.tensor, dtype=torch.uint8): Size is (T, C, H, W) Return: clip (torch.tensor, dtype=torch.fl
interpolation/datasets/video_transforms.py:47
Method__call__
Args: clip (torch.tensor): Video clip to be cropped. Size is (T, C, H, W) Returns: torch.tensor: scale resize
interpolation/datasets/video_transforms.py:79
Method__call__
(self, total_frames)
interpolation/datasets/video_transforms.py:104
Method__init__
( self, vae: AutoencoderKL, text_encoder: CLIPTextModel, tokenizer: CLIPTokeni
base/pipelines/pipeline_videogen.py:99
Method__init__
( self, sample_size: Optional[int] = None, # 64 in_channels: int = 4, out_chan
base/models/unet.py:102
Method__init__
( self, num_attention_heads: int = 16, attention_head_dim: int = 88, in_channe
base/models/transformer_3d.py:76
Method__init__
(self)
base/models/clip.py:25
Method__init__
(self, path, device="cuda", max_length=77)
base/models/clip.py:35
Method__init__
(self, query_dim: int, cross_attention_dim: Optional[int] = None,
base/models/temporal_attention.py:257
Method__init__
( self, heads=8, num_buckets=32, max_distance=128, )
base/models/temporal_attention.py:351
Method__init__
( self, in_channels: int, temb_channels: int, dropout: float = 0.0, nu
base/models/unet_blocks.py:146
Method__init__
( self, in_channels: int, out_channels: int, temb_channels: int, dropo
base/models/unet_blocks.py:236
Method__init__
( self, in_channels: int, out_channels: int, temb_channels: int, dropo
base/models/unet_blocks.py:366
Method__init__
( self, in_channels: int, out_channels: int, prev_output_channel: int,
base/models/unet_blocks.py:445
Method__init__
( self, num_attention_heads: int = 16, attention_head_dim: int = 88, in_channe
base/models/attention.py:296
Method__init__
( self, dim: int, num_attention_heads: int, attention_head_dim: int, d
base/models/attention.py:411
Method__init__
(self, query_dim: int, cross_attention_dim: Optional[int] = None,
base/models/attention.py:563
Method__init__
( self, heads=8, num_buckets=32, max_distance=128, )
base/models/attention.py:670
Method__init__
(self, channels, use_conv=False, out_channels=None, padding=1, name="conv")
base/models/resnet.py:80
Method__init__
( self, *, in_channels, out_channels=None, conv_shortcut=False,
base/models/resnet.py:114
Method__init__
(self, diffusion)
vsr/diffusion/timestep_sampler.py:63
Method__init__
(self, diffusion, history_per_term=10, uniform_prob=0.001)
vsr/diffusion/timestep_sampler.py:121
Method__init__
( self, *, betas, model_mean_type, model_var_type, loss_
vsr/diffusion/gaussian_diffusion.py:146
Method__init__
(self, use_timesteps, **kwargs)
vsr/diffusion/respace.py:73
Method__init__
( self, num_train_timesteps: int = 1000, beta_start: float = 0.0001, beta_end:
vsr/diffusion/scheduling_ddim.py:137
Method__init__
(self)
vsr/models/temporal_module.py:58
Method__init__
( self, in_channels=None, out_channels=None, num_attention_layers=None,
vsr/models/temporal_module.py:66
Method__init__
( self, num_attention_heads=None, attention_head_dim=None, in_channels=None,
vsr/models/temporal_module.py:183
Method__init__
( self, dim=None, num_attention_heads=None, attention_head_dim=None, d
vsr/models/temporal_module.py:307
Method__init__
( self, attention_mode=None, cross_frame_attention_mode=None,
vsr/models/temporal_module.py:431
Method__init__
(self, embedding_dim, num_embeddings)
vsr/models/temporal_module.py:670
Method__init__
( self, heads=8, num_buckets=32, max_distance=128, )
vsr/models/unet.py:58
Method__init__
(self)
vsr/models/clip.py:23
Method__init__
(self, device="cuda", max_length=77)
vsr/models/clip.py:32
Method__init__
( self, in_channels: int = 3, out_channels: int = 3, down_block_types: Tuple[str] = ("DownEncoderBlock
vsr/models/autoencoder_kl.py:77
Method__init__
(self, noise_schedule_config=None)
vsr/models/upscaling.py:28
Method__init__
(self, noise_schedule_config, max_noise_level=1000, to_cuda=False)
vsr/models/upscaling.py:82
Method__init__
( self, in_channels: int, temb_channels: int, dropout: float = 0.0, nu
vsr/models/unet_blocks.py:146
Method__init__
( self, in_channels: int, out_channels: int, temb_channels: int, dropo
vsr/models/unet_blocks.py:236
Method__init__
( self, in_channels: int, out_channels: int, temb_channels: int, dropo
vsr/models/unet_blocks.py:357
Method__init__
( self, in_channels: int, out_channels: int, prev_output_channel: int,
vsr/models/unet_blocks.py:436
Method__init__
( self, num_attention_heads: int = 16, attention_head_dim: int = 88, in_channe
vsr/models/diffusers_attention.py:89
Method__init__
( self, channels: int, num_head_channels: Optional[int] = None, norm_num_group
vsr/models/diffusers_attention.py:267
Method__init__
( self, dim: int, num_attention_heads: int, attention_head_dim: int, d
vsr/models/diffusers_attention.py:401
Method__init__
( self, query_dim: int, cross_attention_dim: Optional[int] = None, heads: int
vsr/models/diffusers_attention.py:527
Method__init__
( self, dim: int, dim_out: Optional[int] = None, mult: int = 4, dropou
vsr/models/diffusers_attention.py:746
Method__init__
(self, dim_in: int, dim_out: int)
vsr/models/diffusers_attention.py:810
Method__init__
(self, dim_in: int, dim_out: int)
vsr/models/diffusers_attention.py:832
Method__init__
(self, embedding_dim, num_embeddings)
vsr/models/diffusers_attention.py:846
Method__init__
( self, num_attention_heads: int = 16, attention_head_dim: int = 88, in_channe
vsr/models/diffusers_attention.py:887
Method__init__
( self, num_attention_heads: int = 16, attention_head_dim: int = 88, in_channe
vsr/models/attention.py:316
Method__init__
( self, dim: int, num_attention_heads: int, attention_head_dim: int, d
vsr/models/attention.py:442
Method__init__
(self, query_dim: int, cross_attention_dim: Optional[int] = None,
vsr/models/attention.py:656
Method__init__
( self, heads=8, num_buckets=32, max_distance=128, )
vsr/models/attention.py:789
Method__init__
( self, vae: AutoencoderKL, text_encoder: CLIPTextModel, tokenizer: CLIPTokeni
vsr/models/pipeline_stable_diffusion_upscale_video_3d.py:72
Method__init__
(self, channels, use_conv=False, out_channels=None, padding=1, name="conv")
vsr/models/resnet.py:90
Method__init__
( self, *, in_channels, out_channels=None, conv_shortcut=False,
vsr/models/resnet.py:124
Method__init__
( self, *, in_channels, out_channels=None, kernel=(3,1,1), con
vsr/models/resnet.py:221
Method__init__
(self, diffusion)
interpolation/diffusion/timestep_sampler.py:63
Method__init__
(self, diffusion, history_per_term=10, uniform_prob=0.001)
interpolation/diffusion/timestep_sampler.py:121
Method__init__
( self, *, betas, model_mean_type, model_var_type, loss_
interpolation/diffusion/gaussian_diffusion.py:153
Method__init__
(self, use_timesteps, **kwargs)
interpolation/diffusion/respace.py:73
Method__init__
(self)
interpolation/datasets/video_transforms.py:44
Method__init__
( self, size, interpolation_mode="bilinear", )
interpolation/datasets/video_transforms.py:64
Method__init__
(self, size)
interpolation/datasets/video_transforms.py:101
Method__init__
( self, sample_size: Optional[int] = None, # 64 in_channels: int = 4, out_chan
interpolation/models/unet.py:62
Method__init__
(self)
interpolation/models/clip.py:25
Method__init__
(self, sd_path, device="cuda", max_length=77)
interpolation/models/clip.py:34
Method__init__
( self, in_channels: int, temb_channels: int, dropout: float = 0.0, nu
interpolation/models/unet_blocks.py:142
Method__init__
( self, in_channels: int, out_channels: int, temb_channels: int, dropo
interpolation/models/unet_blocks.py:230
Method__init__
( self, in_channels: int, out_channels: int, temb_channels: int, dropo
interpolation/models/unet_blocks.py:349
Method__init__
( self, in_channels: int, out_channels: int, prev_output_channel: int,
interpolation/models/unet_blocks.py:428
Method__init__
( self, num_attention_heads: int = 16, attention_head_dim: int = 88, in_channe
interpolation/models/attention.py:346
Method__init__
( self, dim: int, num_attention_heads: int, attention_head_dim: int, d
interpolation/models/attention.py:457
Method__init__
(self, channels, use_conv=False, out_channels=None, padding=1, name="conv")
interpolation/models/resnet.py:80
Method__init__
( self, *, in_channels, out_channels=None, conv_shortcut=False,
interpolation/models/resnet.py:114
Method__len__
(self)
vsr/diffusion/scheduling_ddim.py:461
Method__repr__
(self)
interpolation/datasets/video_transforms.py:56
Method__repr__
(self)
interpolation/datasets/video_transforms.py:90
Method_execution_device
r""" Returns the device on which the pipeline's models will be executed. After calling `pipeline.enable_sequential_cpu_offload()` the
base/pipelines/pipeline_videogen.py:251
Method_execution_device
r""" Returns the device on which the pipeline's models will be executed. After calling `pipeline.enable_sequential_cpu_offload()` the
vsr/models/pipeline_stable_diffusion_upscale_video_3d.py:162
Method_memory_efficient_attention_xformers
(self, query, key, value, attention_mask)
base/models/temporal_attention.py:247
Method_scale_timesteps
(self, t)
vsr/diffusion/respace.py:113
← previousnext →201–300 of 553, ranked by callers