SYMBOL INDEX (268 symbols across 22 files)

FILE: core/attention.py
  class Attention (line 31) | class Attention(nn.Module):
    method __init__ (line 32) | def __init__(
    method forward (line 51) | def forward(self, x: Tensor) -> Tensor:
  class MemEffAttention (line 67) | class MemEffAttention(Attention):
    method forward (line 68) | def forward(self, x: Tensor, attn_bias=None) -> Tensor:
  class CrossAttention (line 87) | class CrossAttention(nn.Module):
    method __init__ (line 88) | def __init__(
    method forward (line 113) | def forward(self, q: Tensor, k: Tensor, v: Tensor) -> Tensor:
  class MemEffCrossAttention (line 137) | class MemEffCrossAttention(CrossAttention):
    method forward (line 138) | def forward(self, q: Tensor, k: Tensor, v: Tensor, attn_bias=None) -> ...

FILE: core/control.py
  function retrieve_timesteps (line 54) | def retrieve_timesteps(
  class ControlNetPipeline (line 97) | class ControlNetPipeline(StableDiffusionControlNetPipeline):
    method pred_x0 (line 99) | def pred_x0(
    method next_step (line 120) | def next_step(
    method __call__ (line 144) | def __call__(

FILE: core/diffuser_utils.py
  class MasaCtrlPipeline (line 22) | class MasaCtrlPipeline(StableDiffusionPipeline):
    method next_step (line 24) | def next_step(
    method step (line 47) | def step(
    method image2latent (line 68) | def image2latent(self, image):
    method latent2image (line 80) | def latent2image(self, latents, return_type='np'):
    method latent2image_grad (line 92) | def latent2image_grad(self, latents):
    method __call__ (line 99) | def __call__(
    method invert (line 201) | def invert(

FILE: core/gs.py
  class GaussianRenderer (line 16) | class GaussianRenderer:
    method __init__ (line 17) | def __init__(self, opt: Options):
    method render (line 31) | def render(self, gaussians, cam_view, cam_view_proj, cam_pos, bg_color...
    method save_ply (line 101) | def save_ply(self, gaussians, path, compatible=True):
    method load_ply (line 154) | def load_ply(self, path, compatible=True):

FILE: core/masactrl.py
  class MutualSelfAttentionControl (line 14) | class MutualSelfAttentionControl(AttentionBase):
    method __init__ (line 20) | def __init__(self, start_step=4, start_layer=10, layer_idx=None, step_...
    method attn_batch (line 41) | def attn_batch(self, q, k, v, sim, attn, is_cross, place_in_unet, num_...
    method forward (line 56) | def forward(self, q, k, v, sim, attn, is_cross, place_in_unet, num_hea...
  class MutualSelfAttention3DControl (line 74) | class MutualSelfAttention3DControl(AttentionBase):
    method __init__ (line 80) | def __init__(self, start_steps=4, start_layer=10, layer_idx=None, step...
    method attn_batch (line 101) | def attn_batch(self, q, k, v, sim, attn, is_cross, place_in_unet, num_...
    method forward (line 116) | def forward(self, q, k, v, sim, attn, is_cross, place_in_unet, num_hea...
  class MutualSelfAttentionControlUnion (line 150) | class MutualSelfAttentionControlUnion(MutualSelfAttentionControl):
    method __init__ (line 151) | def __init__(self, start_step=4, start_layer=10, layer_idx=None, step_...
    method forward (line 164) | def forward(self, q, k, v, sim, attn, is_cross, place_in_unet, num_hea...
  class MutualSelfAttentionControlMask (line 189) | class MutualSelfAttentionControlMask(MutualSelfAttentionControl):
    method __init__ (line 190) | def __init__(self,  start_step=4, start_layer=10, layer_idx=None, step...
    method attn_batch (line 213) | def attn_batch(self, q, k, v, sim, attn, is_cross, place_in_unet, num_...
    method forward (line 238) | def forward(self, q, k, v, sim, attn, is_cross, place_in_unet, num_hea...
  class MutualSelfAttentionControlMaskAuto (line 271) | class MutualSelfAttentionControlMaskAuto(MutualSelfAttentionControl):
    method __init__ (line 272) | def __init__(self, start_step=4, start_layer=10, layer_idx=None, step_...
    method after_step (line 302) | def after_step(self):
    method attn_batch (line 306) | def attn_batch(self, q, k, v, sim, attn, is_cross, place_in_unet, num_...
    method aggregate_cross_attn_map (line 335) | def aggregate_cross_attn_map(self, idx):
    method forward (line 348) | def forward(self, q, k, v, sim, attn, is_cross, place_in_unet, num_hea...

FILE: core/masactrl_utils.py
  class AttentionBase (line 14) | class AttentionBase:
    method __init__ (line 15) | def __init__(self):
    method after_step (line 20) | def after_step(self):
    method __call__ (line 23) | def __call__(self, q, k, v, sim, attn, is_cross, place_in_unet, num_he...
    method forward (line 33) | def forward(self, q, k, v, sim, attn, is_cross, place_in_unet, num_hea...
    method reset (line 38) | def reset(self):
  class AttentionStore (line 43) | class AttentionStore(AttentionBase):
    method __init__ (line 44) | def __init__(self, res=[32], min_step=0, max_step=1000):
    method after_step (line 57) | def after_step(self):
    method forward (line 70) | def forward(self, q, k, v, sim, attn, is_cross, place_in_unet, num_hea...
  function regiter_attention_editor_diffusers (line 79) | def regiter_attention_editor_diffusers(unet, editor: AttentionBase):
  function regiter_attention_editor_ldm (line 147) | def regiter_attention_editor_ldm(model, editor: AttentionBase):

FILE: core/models/transformer_mv2d.py
  function conv_nd (line 34) | def conv_nd(dims, *args, **kwargs):
  function my_repeat (line 54) | def my_repeat(tensor, num_repeats):
  class TransformerMV2DModelOutput (line 65) | class TransformerMV2DModelOutput(BaseOutput):
  class TransformerMV2DModel (line 78) | class TransformerMV2DModel(ModelMixin, ConfigMixin):
    method __init__ (line 107) | def __init__(
    method post_init (line 255) | def post_init(self):
    method post_linear_init (line 283) | def post_linear_init(self):
    method forward (line 308) | def forward(
  class BasicMVTransformerBlock (line 461) | class BasicMVTransformerBlock(nn.Module):
    method __init__ (line 482) | def __init__(
    method set_chunk_feed_forward (line 610) | def set_chunk_feed_forward(self, chunk_size: Optional[int], dim: int):
    method forward (line 615) | def forward(
  class CustomAttention (line 711) | class CustomAttention(Attention):
    method set_use_memory_efficient_attention_xformers (line 712) | def set_use_memory_efficient_attention_xformers(
  class CustomJointAttention (line 720) | class CustomJointAttention(Attention):
    method set_use_memory_efficient_attention_xformers (line 721) | def set_use_memory_efficient_attention_xformers(
  class MVAttnProcessor (line 728) | class MVAttnProcessor:
    method __call__ (line 733) | def __call__(
  class XFormersMVAttnProcessor (line 804) | class XFormersMVAttnProcessor:
    method __call__ (line 809) | def __call__(
  class XFormersJointAttnProcessor (line 904) | class XFormersJointAttnProcessor:
    method __call__ (line 909) | def __call__(
  class JointAttnProcessor (line 992) | class JointAttnProcessor:
    method __call__ (line 997) | def __call__(

FILE: core/models/unet_mv2d_blocks.py
  class IdentityMLP (line 34) | class IdentityMLP(nn.Module):
    method __init__ (line 35) | def __init__(self, size):
    method forward (line 40) | def forward(self, x):
    method init_identity (line 44) | def init_identity(self):
  function get_down_block (line 51) | def get_down_block(
  function get_up_block (line 281) | def get_up_block(
  class UNetMidBlockMV2DCrossAttn (line 515) | class UNetMidBlockMV2DCrossAttn(nn.Module):
    method __init__ (line 516) | def __init__(
    method forward (line 604) | def forward(
  class CrossAttnUpBlockMV2D (line 630) | class CrossAttnUpBlockMV2D(nn.Module):
    method __init__ (line 631) | def __init__(
    method forward (line 720) | def forward(
  class CrossAttnDownBlockMV2D (line 791) | class CrossAttnDownBlockMV2D(nn.Module):
    method __init__ (line 792) | def __init__(
    method forward (line 882) | def forward(

FILE: core/models/unet_mv2d_condition.py
  class UNetMV2DConditionOutput (line 76) | class UNetMV2DConditionOutput(BaseOutput):
  class UNetMV2DConditionModel (line 88) | class UNetMV2DConditionModel(ModelMixin, ConfigMixin, UNet2DConditionLoa...
    method __init__ (line 179) | def __init__(
    method attn_processors (line 634) | def attn_processors(self) -> Dict[str, AttentionProcessor]:
    method set_attn_processor (line 657) | def set_attn_processor(self, processor: Union[AttentionProcessor, Dict...
    method set_default_attn_processor (line 691) | def set_default_attn_processor(self):
    method set_attention_slice (line 697) | def set_attention_slice(self, slice_size):
    method _set_gradient_checkpointing (line 762) | def _set_gradient_checkpointing(self, module, value=False):
    method forward (line 766) | def forward(
    method from_pretrained_2d (line 1063) | def from_pretrained_2d(
    method _load_pretrained_model_2d (line 1394) | def _load_pretrained_model_2d(

FILE: core/models/unet_mv2d_condition_depth.py
  class UNetMV2DConditionOutput (line 76) | class UNetMV2DConditionOutput(BaseOutput):
  class UNetMV2DConditionModel (line 88) | class UNetMV2DConditionModel(ModelMixin, ConfigMixin, UNet2DConditionLoa...
    method __init__ (line 179) | def __init__(
    method attn_processors (line 634) | def attn_processors(self) -> Dict[str, AttentionProcessor]:
    method set_attn_processor (line 657) | def set_attn_processor(self, processor: Union[AttentionProcessor, Dict...
    method set_default_attn_processor (line 691) | def set_default_attn_processor(self):
    method set_attention_slice (line 697) | def set_attention_slice(self, slice_size):
    method _set_gradient_checkpointing (line 762) | def _set_gradient_checkpointing(self, module, value=False):
    method forward (line 766) | def forward(
    method from_pretrained_2d (line 1063) | def from_pretrained_2d(
    method _load_pretrained_model_2d (line 1394) | def _load_pretrained_model_2d(

FILE: core/models/unet_mv2d_condition_depth_diffusion.py
  class IdentityMLP (line 74) | class IdentityMLP(nn.Module):
    method __init__ (line 75) | def __init__(self, size):
    method forward (line 80) | def forward(self, x):
    method init_identity (line 83) | def init_identity(self):
  class UNetMV2DConditionOutput (line 90) | class UNetMV2DConditionOutput(BaseOutput):
  class UNetMV2DConditionModel (line 102) | class UNetMV2DConditionModel(ModelMixin, ConfigMixin, UNet2DConditionLoa...
    method __init__ (line 193) | def __init__(
    method attn_processors (line 648) | def attn_processors(self) -> Dict[str, AttentionProcessor]:
    method set_attn_processor (line 671) | def set_attn_processor(self, processor: Union[AttentionProcessor, Dict...
    method set_default_attn_processor (line 705) | def set_default_attn_processor(self):
    method set_attention_slice (line 711) | def set_attention_slice(self, slice_size):
    method _set_gradient_checkpointing (line 776) | def _set_gradient_checkpointing(self, module, value=False):
    method forward (line 780) | def forward(
    method from_pretrained_2d (line 1080) | def from_pretrained_2d(
    method _load_pretrained_model_2d (line 1413) | def _load_pretrained_model_2d(

FILE: core/models/unet_mv2d_condition_depth_diffusion_test.py
  class IdentityMLP (line 74) | class IdentityMLP(nn.Module):
    method __init__ (line 75) | def __init__(self, size):
    method forward (line 80) | def forward(self, x):
    method init_identity (line 83) | def init_identity(self):
  class UNetMV2DConditionOutput (line 90) | class UNetMV2DConditionOutput(BaseOutput):
  class UNetMV2DConditionModel (line 102) | class UNetMV2DConditionModel(ModelMixin, ConfigMixin, UNet2DConditionLoa...
    method __init__ (line 193) | def __init__(
    method attn_processors (line 648) | def attn_processors(self) -> Dict[str, AttentionProcessor]:
    method set_attn_processor (line 671) | def set_attn_processor(self, processor: Union[AttentionProcessor, Dict...
    method set_default_attn_processor (line 705) | def set_default_attn_processor(self):
    method set_attention_slice (line 711) | def set_attention_slice(self, slice_size):
    method _set_gradient_checkpointing (line 776) | def _set_gradient_checkpointing(self, module, value=False):
    method forward (line 780) | def forward(
    method from_pretrained_2d (line 1080) | def from_pretrained_2d(
    method _load_pretrained_model_2d (line 1414) | def _load_pretrained_model_2d(

FILE: core/models_LGM_compos_diffusion.py
  class LGM (line 20) | class LGM(nn.Module):
    method __init__ (line 21) | def __init__(
    method state_dict (line 68) | def state_dict(self, **kwargs):
    method prepare_default_rays (line 77) | def prepare_default_rays(self, device, elevation=0):
    method forward_gaussians (line 104) | def forward_gaussians(self, images, encoder_hidden_states, data):
    method pred_x0 (line 143) | def pred_x0(
    method encode_prompt (line 164) | def encode_prompt(
    method compute_snr (line 216) | def compute_snr(self, timesteps):
    method forward (line 240) | def forward(self, data, step_ratio=1):

FILE: core/models_LGM_compos_diffusion_validate_inversion_2_masa.py
  class LGM (line 25) | class LGM(nn.Module):
    method __init__ (line 26) | def __init__(
    method state_dict (line 85) | def state_dict(self, **kwargs):
    method prepare_default_rays (line 94) | def prepare_default_rays(self, device, elevation=0, proj_matrix=None):
    method prepare_default_rays_zero123 (line 126) | def prepare_default_rays_zero123(self, device, elevation=0, proj_matri...
    method unet_step (line 163) | def unet_step(
    method forward_gaussians (line 183) | def forward_gaussians(self, images, encoder_hidden_states, data, uncon...
    method pred_x0 (line 231) | def pred_x0(
    method step (line 252) | def step(
    method encode_prompt (line 282) | def encode_prompt(
    method compute_snr (line 334) | def compute_snr(self, timesteps):
    method forward (line 358) | def forward(self, data, step_ratio=1):
    method next_step (line 464) | def next_step(
    method image2latent (line 500) | def image2latent(self, image):
    method invert (line 512) | def invert(
    method validate (line 596) | def validate(self, data, num_inference_steps=30, single_image=True):

FILE: core/options_latents_diffusion.py
  class Options (line 7) | class Options:

FILE: core/provider_Gobjaverse_latent_diffusion_insert.py
  class GobjaverseDataset (line 22) | class GobjaverseDataset(Dataset):
    method _warn (line 24) | def _warn(self):
    method __init__ (line 27) | def __init__(self, opt: Options, training=True):
    method __len__ (line 59) | def __len__(self):
    method __getitem__ (line 63) | def __getitem__(self, idx):

FILE: core/unet_LGM_compos.py
  class MVAttention (line 11) | class MVAttention(nn.Module):
    method __init__ (line 12) | def __init__(
    method forward (line 35) | def forward(self, x):
  class UnetAttention (line 51) | class UnetAttention(nn.Module):
    method __init__ (line 52) | def __init__(
    method post_init (line 78) | def post_init(self):
    method forward (line 82) | def forward(self, x, unet_x):
  class ResnetBlock (line 99) | class ResnetBlock(nn.Module):
    method __init__ (line 100) | def __init__(
    method post_init (line 137) | def post_init(self):
    method forward (line 141) | def forward(self, x, temb=None):
  class DownBlock (line 164) | class DownBlock(nn.Module):
    method __init__ (line 165) | def __init__(
    method forward (line 208) | def forward(self, x, unet_xs=None, temb=None):
  class MidBlock (line 232) | class MidBlock(nn.Module):
    method __init__ (line 233) | def __init__(
    method forward (line 257) | def forward(self, x, temb=None):
  class UpBlock (line 266) | class UpBlock(nn.Module):
    method __init__ (line 267) | def __init__(
    method forward (line 298) | def forward(self, x, xs, temb=None):
  class UNet (line 316) | class UNet(nn.Module):
    method __init__ (line 317) | def __init__(
    method forward (line 386) | def forward(self, x, unet_xss=None, temb=None):

FILE: core/utils.py
  function get_rays (line 10) | def get_rays(pose, h, w, fovy, opengl=True):
  function orbit_camera_jitter (line 45) | def orbit_camera_jitter(poses, strength=0.1):
  function grid_distortion (line 63) | def grid_distortion(images, strength=0.5):

FILE: infer_ours_masa.py
  function process (line 72) | def process(opt: Options, path):

FILE: main_resume_compose.py
  function main (line 17) | def main():

FILE: mvdream/mv_unet.py
  function get_camera (line 20) | def get_camera(
  function timestep_embedding (line 42) | def timestep_embedding(timesteps, dim, max_period=10000, repeat_only=Fal...
  function zero_module (line 70) | def zero_module(module):
  function conv_nd (line 79) | def conv_nd(dims, *args, **kwargs):
  function avg_pool_nd (line 92) | def avg_pool_nd(dims, *args, **kwargs):
  function default (line 105) | def default(val, d):
  class GEGLU (line 111) | class GEGLU(nn.Module):
    method __init__ (line 112) | def __init__(self, dim_in, dim_out):
    method forward (line 116) | def forward(self, x):
  class FeedForward (line 121) | class FeedForward(nn.Module):
    method __init__ (line 122) | def __init__(self, dim, dim_out=None, mult=4, glu=False, dropout=0.0):
    method forward (line 136) | def forward(self, x):
  class MemoryEfficientCrossAttention (line 140) | class MemoryEfficientCrossAttention(nn.Module):
    method __init__ (line 142) | def __init__(
    method forward (line 176) | def forward(self, x, context=None):
  class BasicTransformerBlock3D (line 230) | class BasicTransformerBlock3D(nn.Module):
    method __init__ (line 232) | def __init__(
    method forward (line 267) | def forward(self, x, context=None, num_frames=1):
  class SpatialTransformer3D (line 276) | class SpatialTransformer3D(nn.Module):
    method __init__ (line 278) | def __init__(
    method forward (line 318) | def forward(self, x, context=None, num_frames=1):
  class PerceiverAttention (line 335) | class PerceiverAttention(nn.Module):
    method __init__ (line 336) | def __init__(self, *, dim, dim_head=64, heads=8):
    method forward (line 350) | def forward(self, x, latents):
  class Resampler (line 386) | class Resampler(nn.Module):
    method __init__ (line 387) | def __init__(
    method forward (line 420) | def forward(self, x):
  class CondSequential (line 431) | class CondSequential(nn.Sequential):
    method forward (line 437) | def forward(self, x, emb, context=None, num_frames=1):
  class Upsample (line 448) | class Upsample(nn.Module):
    method __init__ (line 457) | def __init__(self, channels, use_conv, dims=2, out_channels=None, padd...
    method forward (line 468) | def forward(self, x):
  class Downsample (line 481) | class Downsample(nn.Module):
    method __init__ (line 490) | def __init__(self, channels, use_conv, dims=2, out_channels=None, padd...
    method forward (line 510) | def forward(self, x):
  class ResBlock (line 515) | class ResBlock(nn.Module):
    method __init__ (line 530) | def __init__(
    method forward (line 592) | def forward(self, x, emb):
  class MultiViewUNetModel (line 615) | class MultiViewUNetModel(ModelMixin, ConfigMixin):
    method __init__ (line 645) | def __init__(
    method forward (line 944) | def forward(

FILE: mvdream/pipeline_mvdream.py
  class MVDreamPipeline (line 23) | class MVDreamPipeline(DiffusionPipeline):
    method __init__ (line 27) | def __init__(
    method enable_vae_slicing (line 84) | def enable_vae_slicing(self):
    method disable_vae_slicing (line 93) | def disable_vae_slicing(self):
    method enable_vae_tiling (line 100) | def enable_vae_tiling(self):
    method disable_vae_tiling (line 109) | def disable_vae_tiling(self):
    method enable_sequential_cpu_offload (line 116) | def enable_sequential_cpu_offload(self, gpu_id=0):
    method enable_model_cpu_offload (line 140) | def enable_model_cpu_offload(self, gpu_id=0):
    method _execution_device (line 170) | def _execution_device(self):
    method _encode_prompt (line 187) | def _encode_prompt(
    method decode_latents (line 339) | def decode_latents(self, latents):
    method prepare_extra_step_kwargs (line 347) | def prepare_extra_step_kwargs(self, generator, eta):
    method prepare_latents (line 368) | def prepare_latents(
    method encode_image (line 402) | def encode_image(self, image, device, num_images_per_prompt):
    method encode_image_latents (line 416) | def encode_image_latents(self, image, device, num_images_per_prompt):
    method __call__ (line 432) | def __call__(