MCPcopy Create free account

hub / github.com/CompVis/zigma / functions

Functions471 in github.com/CompVis/zigma

⨍ Functions 471◇ Types & classes 98
Functionnone_or_str
(value)
utils/train_utils_args.py:6
Methodoperator()
dis_causal_conv1d/csrc/causal_conv1d_common.h:43
Methodoperator()
dis_mamba/csrc/selective_scan/selective_scan_common.h:112
Methodoperator()
dis_mamba/csrc/selective_scan/selective_scan_common.h:119
Methodoperator()
Callback operator to be entered by the first warp of threads in the block. Thread-0 is responsible for returning a value for seeding the block-wide sc
dis_mamba/csrc/selective_scan/selective_scan_common.h:139
Functionoperator+
dis_mamba/csrc/selective_scan/selective_scan_common.h:15
Functionparse_ode_args
(parser)
utils/train_utils_args.py:19
Functionparse_sde_args
(parser)
utils/train_utils_args.py:27
Functionparse_transport_args
(parser)
utils/train_utils_args.py:11
Methodplot
Plot a single or multiple values from the metric. Args: val: Either a single result from calling `metric.forward` or `metric.comp
utils/torchmetric_sfid.py:438
Methodplot
Plot a single or multiple values from the metric. Args: val: Either a single result from calling `metric.forward` or `metric.comp
utils/torchmetric_fvd.py:445
Methodplot
Plot a single or multiple values from the metric. Args: val: Either a single result from calling `metric.forward` or `metric.comp
utils/torchmetric_inception.py:175
Functionrand_perm
(N, num)
utils/utils_zigzag.py:178
Methodreplace_logits
(self, num_classes)
video_metrics/fvd/videogpt/pytorch_i3d.py:290
Methodreset
Reset metric states.
utils/torchmetric_sfid.py:413
Methodreset
Reset metric states.
utils/torchmetric_fdd.py:342
Methodreset
(self)
utils/torchmetric_kid.py:275
Methodreset
Reset metric states.
utils/torchmetric_fvd.py:420
Methodreset
(self)
utils/torchmetric_prdc.py:196
Functionresized_crop
Do spatial cropping and resizing to the video clip Args: clip (torch.tensor): Video clip to be cropped. Size is (T, C, H, W)
datasets/video_utils.py:80
Methodresume
(self, ckpt_root, step=None)
utils/train_state_utils.py:48
Functionrms_norm_ref
(x, weight, bias, residual=None, eps=1e-6, prenorm=False, upcast=False)
dis_mamba/mamba_ssm/ops/triton/layernorm.py:35
Methodrun
(self)
dis_causal_conv1d/setup.py:199
Methodrun
dis_causal_conv1d/csrc/causal_conv1d_common.h:50
Methodrun
dis_causal_conv1d/csrc/causal_conv1d_common.h:60
Methodsample
(self, x, model, **model_kwargs)
transport/integrators.py:105
Methodscore_ode
(x, t, model, **model_kwargs)
transport/transport.py:178
Functionselective_scan_bwd
dis_mamba/csrc/selective_scan/selective_scan.cpp:338
Functionselective_scan_flop_jit
(inputs, outputs)
model_zigma.py:1173
Functionselective_scan_fwd
dis_mamba/csrc/selective_scan/selective_scan.cpp:226
Methodset_dtype
Transfer all metric state to specific dtype. Special version of standard `type` method. Arguments: dst_type: the desired type as
utils/torchmetric_sfid.py:426
Methodset_dtype
Transfer all metric state to specific dtype. Special version of standard `type` method. Arguments: dst_type: the desired type as
utils/torchmetric_fdd.py:355
Methodset_dtype
Transfer all metric state to specific dtype. Special version of standard `type` method. Arguments: dst_type: the desired type as
utils/torchmetric_fvd.py:433
Methodsetup
(self, arch=None, clean_resize: bool = False)
utils/torchmetric_fdd.py:83
Functionstore_output
dis_mamba/csrc/selective_scan/selective_scan_common.h:204
Functiont2i_modulate
(x, shift, scale)
model_zigma.py:57
Functiontest_bimamba_inner_fn
(is_variable_B, is_variable_C, seqlen, itype, wtype)
dis_mamba/tests/ops/test_selective_scan.py:265
Functiontest_causal_conv1d
(seqlen, width, has_bias, silu_activation, itype, channel_last)
dis_causal_conv1d/tests/test_causal_conv1d.py:29
Functiontest_causal_conv1d_race_condition
(seqlen, width, has_bias, silu_activation, itype, channel_last)
dis_causal_conv1d/tests/test_causal_conv1d.py:133
Functiontest_causal_conv1d_update
(dim, width, has_bias, silu_activation, itype)
dis_causal_conv1d/tests/test_causal_conv1d.py:88
Functiontest_causal_conv1d_update
(dim, dstate, has_z, itype)
dis_mamba/tests/ops/triton/test_selective_state_update.py:22
Methodtest_dataloader
(self)
datasets/wds_dataloader.py:188
Functiontest_hibert_fig
(num_dims=2, img_size_power=3)
utils/utils_zigzag.py:207
Functiontest_mamba_inner_fn
(is_variable_B, is_variable_C, seqlen, itype, wtype)
dis_mamba/tests/ops/test_selective_scan.py:162
Functiontest_sd_1_5
()
wandb_utils.py:115
Functiontest_selective_scan
(is_variable_B, is_variable_C, varBC_groups, has_D, has_z, has_delta_bias, delta_softp
dis_mamba/tests/ops/test_selective_scan.py:40
Functiontest_webdataset_faceshq_sd1_5
()
wandb_utils.py:128
Methodto_float
dis_mamba/csrc/selective_scan/selective_scan_common.h:60
Methodto_float
dis_mamba/csrc/selective_scan/selective_scan_common.h:68
Methodto_float
dis_mamba/csrc/selective_scan/selective_scan_common.h:80
Methodtransform
(self, image)
utils/torchmetric_fdd.py:93
Functiontraverse_grid_v1_continuous
(N)
utils/utils_zigzag.py:190
Functionuniq
(arr)
model_zigma.py:85
Methodupdate
Update the state with extracted features. Args: imgs: tensor with images feed to the feature extractor real: bool ind
utils/torchmetric_kid.py:233
Methodupdate
Update the state with extracted features.
utils/torchmetric_inception.py:145
Methodval_dataloader
(self)
datasets/wds_dataloader.py:185
Methodvelocity_ode
(x, t, model, **model_kwargs)
transport/transport.py:190
Functionzigma_b_1
(**kwargs)
model_zigma.py:1085
Functionzigma_b_2
(**kwargs)
model_zigma.py:1090
Functionzigma_b_4
(**kwargs)
model_zigma.py:1095
Functionzigma_h_1
(**kwargs)
model_zigma.py:1125
Functionzigma_h_2
(**kwargs)
model_zigma.py:1130
Functionzigma_h_4
(**kwargs)
model_zigma.py:1135
Functionzigma_l_1
(**kwargs)
model_zigma.py:1100
Functionzigma_l_2
(**kwargs)
model_zigma.py:1105
Functionzigma_l_4
(**kwargs)
model_zigma.py:1110
Functionzigma_m_2
(**kwargs)
model_zigma.py:1115
Functionzigma_m_4
(**kwargs)
model_zigma.py:1120
Functionzigma_s_1
(**kwargs)
model_zigma.py:1075
Functionzigma_s_2
(**kwargs)
model_zigma.py:1070
Functionzigma_s_4
(**kwargs)
model_zigma.py:1080
← previous401–471 of 471, ranked by callers