MCPcopy Create free account

hub / github.com/baaivision/GeoDream / functions

Functions3,158 in github.com/baaivision/GeoDream

⨍ Functions 3,158◇ Types & classes 664↳ Endpoints 1
↓ 441 callersMethodto
Same as to in torch module Don't really underestand why this isn't a module in the first place
mv-diffusion/One-2-3-45-by-view/ldm/models/diffusion/ddim.py:21
↓ 319 callersMethodto
Same as to in torch module Don't really underestand why this isn't a module in the first place
extern/ldm_zero123/models/diffusion/ddim.py:30
↓ 118 callersMethodC
(self, value: Any)
threestudio/systems/base.py:92
↓ 83 callersMethodinterpolate
( self, attr: Float[Tensor, "B Nv C"], rast: Float[Tensor, "B H W 4"], tri: In
threestudio/utils/rasterize.py:58
↓ 74 callersMethodregister_buffer
(self, name, attr)
extern/ldm_zero123/models/diffusion/ddim.py:38
↓ 57 callersMethodeval
(self)
threestudio/models/renderers/patch_renderer.py:105
↓ 56 callersMethodfrom_pretrained
(cls, name="vgg_lpips")
threestudio/utils/perceptual/perceptual.py:36
↓ 53 callersMethodload
(cls, path: str, arr_name: str)
mv-diffusion/One-2-3-45-by-view/ldm/modules/evaluate/adm_evaluator.py:521
↓ 53 callersMethodregister_buffer
(self, name, attr, device=None)
mv-diffusion/One-2-3-45-by-view/ldm/models/diffusion/ddim.py:30
↓ 40 callersMethodload
(cls, path: str, arr_name: str)
extern/ldm_zero123/modules/evaluate/adm_evaluator.py:548
↓ 38 callersMethodload_state_dict
(network, checkpoint, comment)
mv-diffusion/One-2-3-45/reconstruction/exp_runner_generic_blender_val.py:437
↓ 34 callersMethodsave_image_grid
( self, filename, imgs, align=DEFAULT_GRID_KWARGS["align"], name: Opti
threestudio/utils/saving.py:301
↓ 31 callersMethodregister_buffer
(self, name, attr)
mv-diffusion/MVDream/mvdream/ldm/models/diffusion/ddim.py:18
↓ 29 callersMethodwrite
(self, msg: str)
threestudio/utils/callbacks.py:134
↓ 28 callersFunctionC
(value: Any, epoch: int, global_step: int)
threestudio/utils/misc.py:65
↓ 23 callersFunctioninstantiate_from_config
(config)
mv-diffusion/One-2-3-45-by-view/ldm/util.py:131
↓ 22 callersMethodmeshgrid
(self, h, w)
mv-diffusion/One-2-3-45-by-view/ldm/models/diffusion/ddpm.py:632
↓ 21 callersMethoddecode_first_stage
(self, z, predict_cids=False, force_not_quantize=False)
mv-diffusion/One-2-3-45-by-view/ldm/models/diffusion/ddpm.py:763
↓ 20 callersMethoddecode_first_stage
(self, z, predict_cids=False, force_not_quantize=False)
extern/ldm_zero123/models/diffusion/ddpm.py:984
↓ 19 callersMethodconvert_data
(self, data)
threestudio/utils/saving.py:34
↓ 19 callersMethodstep
Performs a single optimization step.
threestudio/systems/optimizers.py:104
↓ 18 callersMethodget_save_path
(self, filename)
threestudio/utils/saving.py:51
↓ 17 callersFunctionextract_into_tensor
(a, t, x_shape)
extern/ldm_zero123/modules/diffusionmodules/util.py:119
↓ 17 callersFunctionextract_into_tensor
(a, t, x_shape)
mv-diffusion/One-2-3-45-by-view/ldm/modules/diffusionmodules/util.py:96
↓ 17 callersFunctionget_activation
(name)
threestudio/utils/ops.py:77
↓ 17 callersMethodget_text_embeddings
( self, prompt: Union[str, List[str]], negative_prompt: Union[str, List[str]] )
threestudio/models/prompt_processors/base.py:465
↓ 16 callersFunctioncleanup
()
threestudio/utils/misc.py:89
↓ 16 callersFunctioncontract_to_unisphere
( x: Float[Tensor, "... 3"], bbox: Float[Tensor, "2 3"], unbounded: bool = False )
threestudio/models/geometry/base.py:20
↓ 16 callersFunctionconv_nd
Create a 1D, 2D, or 3D convolution module.
mv-diffusion/MVDream/mvdream/ldm/modules/diffusionmodules/util.py:233
↓ 16 callersMethoddecode
(self, quant)
extern/ldm_zero123/models/autoencoder.py:117
↓ 16 callersFunctiondot
(x, y)
threestudio/utils/ops.py:15
↓ 16 callersFunctionexists
(val)
extern/ldm_zero123/modules/x_transformer.py:53
↓ 16 callersFunctionexists
(val)
mv-diffusion/One-2-3-45-by-view/ldm/modules/x_transformer.py:54
↓ 16 callersFunctioninstantiate_from_config
(config)
extern/ldm_zero123/util.py:98
↓ 16 callersFunctionparse_version
(ver: str)
threestudio/utils/misc.py:13
↓ 16 callersMethodsave_img_sequence
( self, filename, img_dir, matcher, save_format="mp4", fps=30,
threestudio/utils/saving.py:395
↓ 15 callersMethod__init__
( self, *, ch, out_ch, ch_mult=(1, 2, 4, 8), num_res_blocks,
extern/ldm_zero123/modules/diffusionmodules/model.py:206
↓ 15 callersMethod__init__
( self, n_embed, n_layer, vocab_size=30522, max_seq_len=77, de
extern/ldm_zero123/modules/encoders/modules.py:180
↓ 15 callersMethod__init__
(self, *, ch, out_ch, ch_mult=(1,2,4,8), num_res_blocks, attn_resolutions, dropout=0.0, resam
mv-diffusion/One-2-3-45-by-view/ldm/modules/diffusionmodules/model.py:217
↓ 15 callersMethod__init__
(self, n_embed, n_layer, vocab_size=30522, max_seq_len=77, device="cuda",use_tokenizer=True,
mv-diffusion/One-2-3-45-by-view/ldm/modules/encoders/modules.py:152
↓ 15 callersMethod__init__
( self, *, ch, out_ch, ch_mult=(1, 2, 4, 8), num_res_blocks,
threestudio/utils/GAN/vae.py:205
↓ 15 callersFunctionconv_nd
Create a 1D, 2D, or 3D convolution module.
extern/ldm_zero123/modules/diffusionmodules/util.py:246
↓ 15 callersFunctionconv_nd
Create a 1D, 2D, or 3D convolution module.
mv-diffusion/One-2-3-45-by-view/ldm/modules/diffusionmodules/util.py:218
↓ 15 callersMethoddecode
(self, quant)
mv-diffusion/One-2-3-45-by-view/ldm/models/autoencoder.py:107
↓ 14 callersMethod__init__
(self, *, ch, out_ch, ch_mult=(1,2,4,8), num_res_blocks, attn_resolutions, dropout=0.0, resam
mv-diffusion/MVDream/mvdream/ldm/modules/diffusionmodules/model.py:301
↓ 14 callersFunctionget_mlp
(n_input_dims, n_output_dims, config)
threestudio/models/networks.py:272
↓ 14 callersMethodq_sample
(self, x_start, t, noise=None)
extern/ldm_zero123/models/diffusion/ddpm.py:425
↓ 14 callersMethodq_sample
(self, x_start, t, noise=None)
mv-diffusion/One-2-3-45-by-view/ldm/models/diffusion/ddpm.py:324
↓ 13 callersFunctionassign_to_checkpoint
This does the final conversion step: take locally converted weights and apply a global renaming to them. It splits attention layers, and take
scripts/convert_zero123_to_diffusers.py:137
↓ 13 callersFunctionexists
(val)
mv-diffusion/MVDream/mvdream/ldm/modules/attention.py:23
↓ 13 callersMethodregister_buffer
(self, name, attr)
extern/ldm_zero123/models/diffusion/plms.py:24
↓ 13 callersMethodregister_buffer
(self, name, attr)
mv-diffusion/One-2-3-45-by-view/ldm/models/diffusion/plms.py:19
↓ 13 callersMethodsample_log
(self, cond, batch_size, ddim, ddim_steps, **kwargs)
extern/ldm_zero123/models/diffusion/ddpm.py:1633
↓ 13 callersMethodsample_log
(self, cond, batch_size, ddim, ddim_steps, **kwargs)
mv-diffusion/One-2-3-45-by-view/ldm/models/diffusion/ddpm.py:1227
↓ 12 callersMethod__init__
(self, value, fn)
extern/ldm_zero123/modules/x_transformer.py:125
↓ 12 callersMethod__init__
(self, value, fn)
mv-diffusion/One-2-3-45-by-view/ldm/modules/x_transformer.py:118
↓ 12 callersFunctionbinary_cross_entropy
F.binary_cross_entropy is not numerically stable in mixed-precision training.
threestudio/utils/ops.py:295
↓ 12 callersFunctionchunk_batch
(func: Callable, chunk_size: int, *args, **kwargs)
threestudio/utils/ops.py:112
↓ 12 callersMethoddecode
( self, x_latent, cond, t_start, unconditional_guidance_scale=1.0,
extern/ldm_zero123/models/diffusion/ddim.py:453
↓ 12 callersMethoddecode
(self, x_latent, cond, t_start, unconditional_guidance_scale=1.0, unconditional_conditioning=None,
mv-diffusion/One-2-3-45-by-view/ldm/models/diffusion/ddim.py:308
↓ 12 callersFunctionexists
(x)
extern/ldm_zero123/util.py:73
↓ 12 callersFunctionexists
(x)
mv-diffusion/One-2-3-45-by-view/ldm/util.py:106
↓ 12 callersFunctionextract_into_tensor
(a, t, x_shape)
mv-diffusion/MVDream/mvdream/ldm/modules/diffusionmodules/util.py:111
↓ 12 callersFunctionget_block
(in_channel, depth, num_units, stride=2)
extern/ldm_zero123/thirdp/psp/helpers.py:38
↓ 12 callersFunctionget_block
(in_channel, depth, num_units, stride=2)
mv-diffusion/One-2-3-45-by-view/ldm/thirdp/psp/helpers.py:27
↓ 12 callersMethodget_input
(self, batch, k)
extern/ldm_zero123/models/diffusion/ddpm.py:487
↓ 12 callersMethodget_input
(self, batch, k)
mv-diffusion/One-2-3-45-by-view/ldm/models/diffusion/ddpm.py:379
↓ 12 callersMethodload_state_dict
(network, checkpoint, comment)
mv-diffusion/One-2-3-45-by-view/reconstruction/exp_runner_generic_blender_val.py:438
↓ 12 callersFunctionparse_structured
(fields: Any, cfg: Optional[Union[dict, DictConfig]] = None)
threestudio/utils/config.py:121
↓ 11 callersMethodapply_model
(self, x_noisy, t, cond, return_ids=False)
extern/ldm_zero123/models/diffusion/ddpm.py:1130
↓ 11 callersMethodencode
(self, c)
threestudio/utils/GAN/discriminator.py:114
↓ 11 callersMethodexport
(self, *args, **kwargs)
threestudio/models/geometry/base.py:50
↓ 11 callersFunctionlinear
Create a linear module.
mv-diffusion/MVDream/mvdream/ldm/modules/diffusionmodules/util.py:246
↓ 11 callersFunctionlog_txt_as_img
(wh, xc, size=10)
extern/ldm_zero123/util.py:35
↓ 11 callersFunctionlog_txt_as_img
(wh, xc, size=10)
mv-diffusion/One-2-3-45-by-view/ldm/util.py:70
↓ 10 callersMethod__init__
(self, channels, use_conv, dims=2, out_channels=None, padding=1)
mv-diffusion/MVDream/mvdream/ldm/modules/diffusionmodules/openaimodel.py:100
↓ 10 callersFunctionconfig_to_primitive
(config, resolve: bool = True)
threestudio/utils/config.py:112
↓ 10 callersFunctioncubic_interpolate
one dimensional cubic interpolation :param p: [N, 4] (4) should be in order :param x: [N] :return:
mv-diffusion/One-2-3-45-by-view/reconstruction/ops/grid_sampler.py:241
↓ 10 callersFunctioncubic_interpolate
one dimensional cubic interpolation :param p: [N, 4] (4) should be in order :param x: [N] :return:
mv-diffusion/One-2-3-45/reconstruction/ops/grid_sampler.py:241
↓ 10 callersFunctioncubic_interpolate
one dimensional cubic interpolation :param p: [N, 4] (4) should be in order :param x: [N] :return:
threestudio/models/geometry/grid_sampler.py:241
↓ 10 callersFunctionget_rank
()
threestudio/utils/misc.py:17
↓ 10 callersFunctionnonlinearity
(x)
extern/ldm_zero123/modules/diffusionmodules/model.py:34
↓ 10 callersFunctionnonlinearity
(x)
mv-diffusion/One-2-3-45-by-view/ldm/modules/diffusionmodules/model.py:33
↓ 10 callersFunctionnonlinearity
(x)
threestudio/utils/GAN/vae.py:35
↓ 9 callersFunctionNormalize
(in_channels, num_groups=32)
extern/ldm_zero123/modules/diffusionmodules/model.py:39
↓ 9 callersFunctionNormalize
(in_channels, num_groups=32)
mv-diffusion/MVDream/mvdream/ldm/modules/diffusionmodules/model.py:46
↓ 9 callersFunctionNormalize
(in_channels, num_groups=32)
mv-diffusion/One-2-3-45-by-view/ldm/modules/diffusionmodules/model.py:38
↓ 9 callersFunctionNormalize
(in_channels, num_groups=32)
threestudio/utils/GAN/vae.py:40
↓ 9 callersMethod__init__
(self, channels, use_conv, dims=2, out_channels=None, padding=1)
extern/ldm_zero123/modules/diffusionmodules/openaimodel.py:104
↓ 9 callersMethod__init__
(self, channels, use_conv, dims=2, out_channels=None, padding=1)
mv-diffusion/One-2-3-45-by-view/ldm/modules/diffusionmodules/openaimodel.py:101
↓ 9 callersMethod__init__
(self, dim_in: int, dim_out: int, config: dict)
threestudio/models/networks.py:151
↓ 9 callersMethodenable_sequential_cpu_offload
r""" Offloads all models to CPU using accelerate, significantly reducing memory usage. When called, unet, text_encoder, vae and safety
extern/zero123.py:180
↓ 9 callersFunctionnonlinearity
(x)
mv-diffusion/MVDream/mvdream/ldm/modules/diffusionmodules/model.py:41
↓ 9 callersFunctionsample_ptsFeatures_from_featureMaps
sample features of pts from 2d feature maps :param pts: [N_rays, N_samples, 3] :param featureMaps: [N_views, C, H, W] :param w2cs: [N
mv-diffusion/One-2-3-45-by-view/reconstruction/models/render_utils.py:88
↓ 9 callersFunctionsample_ptsFeatures_from_featureMaps
sample features of pts from 2d feature maps :param pts: [N_rays, N_samples, 3] :param featureMaps: [N_views, C, H, W] :param w2cs: [N
mv-diffusion/One-2-3-45/reconstruction/models/render_utils.py:88
↓ 9 callersMethodsampling
Sampling with CDFs from proposal networks. Args: prop_sigma_fns: Proposal network evaluate functions. It should be a list
threestudio/models/estimators.py:23
↓ 9 callersFunctionshift_azimuth_deg
(azimuth: Float[Tensor, "..."])
threestudio/models/prompt_processors/base.py:168
↓ 9 callersMethodunet
(self)
threestudio/models/guidance/sd_vsd_scale_guidance.py:241
↓ 8 callersMethod__init__
(self, **kwargs)
mv-diffusion/One-2-3-45-by-view/reconstruction/tsparse/modules.py:157
↓ 8 callersMethod__init__
(self, **kwargs)
mv-diffusion/One-2-3-45/reconstruction/tsparse/modules.py:157
next →1–100 of 3,158, ranked by callers