MCPcopy Create free account

hub / github.com/StargazerX0/ScaleKV / functions

Functions369 in github.com/StargazerX0/ScaleKV

⨍ Functions 369◇ Types & classes 52
Method__len__
(self)
scripts/compute_metrics.py:39
Method__repr__
(self)
infinity/dataset/build.py:75
Method__str__
(self)
infinity/utils/misc.py:205
Method__str__
(self)
infinity/utils/misc.py:232
Method__str__
(self)
infinity/utils/arg_util.py:345
Methodadd_meter
(self, name, meter)
infinity/utils/misc.py:245
Methodadd_model_specific_args
(parent_parser)
infinity/models/bsq_vae/flux_vqgan.py:548
Functionallgather
(t: torch.Tensor, cat=True)
infinity/utils/dist.py:143
Functionallgather_diff_shape
(t: torch.Tensor, cat=True)
infinity/utils/dist.py:156
Functionauto_resume
(args: arg_util.Args, pattern='ckpt*.pth')
infinity/utils/save_and_load.py:112
Methodauto_sync
(source_filename, target_filename)
infinity/utils/save_and_load.py:90
Methodavg
(self)
infinity/utils/misc.py:186
Methodbackward_clip_step
( self, ep: int, it: int, g_it: int, stepping: bool, logging_params: bool, loss: torch.Tensor, clip_de
infinity/utils/amp_opt.py:111
Functionbroadcast
(t: torch.Tensor, src_rank)
infinity/utils/dist.py:183
Functionbuild_t2i_dataset
( args, data_path: str, data_load_reso: int, max_caption_len: int, short_prob=0.2, loa
infinity/dataset/build.py:122
Functionbuild_vae_gpt
(args: arg_util.Args, vae_st: dict, skip_gpt: bool, force_flash=False, device='cuda')
infinity/utils/load.py:22
Functioncalculate_frechet_distance
Numpy implementation of the Frechet Distance. The Frechet distance between two multivariate Gaussians X_1 ~ N(mu_1, C_1) and X_2 ~ N(mu_2, C_2
tools/inception.py:19
Functioncenter_crop_arr
Center cropping implementation from ADM. https://github.com/openai/guided-diffusion/blob/8fb3ad9197f16bbc40620447b2742e13458d2831/guided_diff
infinity/dataset/build.py:45
Functioncenter_crop_to_tensor_pm1
Center cropping implementation from ADM. https://github.com/openai/guided-diffusion/blob/8fb3ad9197f16bbc40620447b2742e13458d2831/guided_diff
infinity/dataset/dataset_t2i_iterable.py:30
Functioncheck_randomness
(args)
infinity/utils/misc.py:386
Methodclean_caption
(self, caption)
infinity/models/t5.py:146
Methodclean_caption_simplify
(self, caption)
infinity/models/t5.py:257
Methodcodebooks
(self)
infinity/models/bsq_vae/multiscale_bsq.py:245
Methodcollate_function
(batch, online_t5: bool = False)
infinity/dataset/dataset_t2i_iterable.py:363
Methodcompile_model
(self, m, fast)
infinity/utils/arg_util.py:270
Methoddbg_ks_this_line
(self, g_it: int)
infinity/utils/arg_util.py:205
Functiondenormalize_pm1_into_01
(x)
infinity/dataset/build.py:41
Functiondist_fmt_vals
(val: float, fmt: Union[str, None] = '%.2f')
infinity/utils/dist.py:193
Methoddo_nothing
(*args, **kwargs)
infinity/utils/misc.py:51
Methoddump_log
(self)
infinity/utils/arg_util.py:279
Methodearly_stop
(self)
infinity/utils/misc.py:109
Functionecho
(info)
infinity/utils/misc.py:22
Functionencode_image
Resize an image and encode it as a Base64 string. Args: - image_path (str): Path to the image file. - size (tuple): New size as
tools/prompt_rewriter.py:21
Functionenhance_image
(image)
tools/run_infinity.py:67
Functionentropy
(prob)
infinity/models/bsq_vae/multiscale_bsq.py:82
Methodextra_repr
(self)
infinity/models/flex_attn.py:128
Methodextra_repr
(self)
infinity/models/infinity.py:714
Methodextra_repr
(self)
infinity/models/basic.py:134
Methodextra_repr
(self)
infinity/models/basic.py:172
Methodextra_repr
(self)
infinity/models/basic.py:191
Methodextra_repr
(self)
infinity/models/basic.py:313
Methodextra_repr
(self)
infinity/models/basic.py:403
Methodextra_repr
(self)
infinity/models/basic.py:450
Methodextra_repr
(self)
infinity/models/basic.py:521
Functionextract_key_val
(text)
tools/run_infinity.py:30
Functionfeatures2image
(raw_features)
infinity/models/bitwise_self_correction.py:16
Functionfilter_params
(model, ndim_dict, nowd_keys=(), lr_scale=0.0)
infinity/utils/lr_control.py:67
Functionfinalize
()
infinity/utils/dist.py:243
Methodfinishing
(self)
infinity/utils/misc.py:130
Methodflip_requant
(self, vae_scale_schedule, inp_B3HW, raw_features, device)
infinity/models/bitwise_self_correction.py:32
Functionfor_visualize
(func)
infinity/utils/dist.py:231
Methodforward
Get Inception feature maps Parameters ---------- inp : torch.autograd.Variable Input tensor of shape Bx3xHxW. Val
tools/inception.py:188
Methodforward
(self, x)
tools/inception.py:258
Methodforward
(self, x)
tools/inception.py:283
Methodforward
(self, x)
tools/inception.py:311
Methodforward
(self, x)
tools/inception.py:344
Methodforward
(self, *args, **kwargs)
infinity/utils/misc.py:296
Methodforward
(self, q, k, v, scale = None)
infinity/models/flex_attn.py:114
Methodforward
(self, x, *args, **kwargs)
infinity/models/infinity.py:34
Methodforward
(self, ca_kv)
infinity/models/infinity.py:49
Methodforward
(self, cond_BD)
infinity/models/infinity.py:53
Methodforward
(self, x, cond_BD, ca_kv, attn_bias_or_two_vector, attn_fn=None, scale_schedule=None, checkpointing_full_block
infinity/models/infinity.py:65
Methodforward
(self, x)
infinity/models/basic.py:130
Methodforward
(self, x)
infinity/models/basic.py:154
Methodforward
(self, x)
infinity/models/basic.py:188
Methodforward
:param (fp32) x: shaped (B or batch_size, L or seq_length, C or hidden_dim); if seq-parallel is used, the `L` dim would be shared :pa
infinity/models/basic.py:248
Methodforward
:param q: shaped as (batch, seq_len, Q_dim) :param ca_kv: contains several vectors, each of which is shaped as (len_i, KV_dim). We ha
infinity/models/basic.py:359
Methodforward
(self, x, cond_BD, ca_kv, attn_bias_or_two_vector)
infinity/models/basic.py:435
Methodforward
(self, x, cond_BD, ca_kv, attn_bias_or_two_vector, attn_fn=None, scale_schedule=None, rope2d_freqs_grid=None,
infinity/models/basic.py:494
Methodforward
(self, x_BLC: torch.Tensor, cond_BD: Optional[torch.Tensor])
infinity/models/basic.py:535
Methodforward
(self, x)
infinity/models/bsq_vae/multiscale_bsq.py:98
Methodforward
(self, x)
infinity/models/bsq_vae/multiscale_bsq.py:155
Methodforward
( self, x, scale_schedule=None, mask = None, return_all_codes = False,
infinity/models/bsq_vae/multiscale_bsq.py:271
Methodforward
einstein notation b - batch n - sequence (or flattened spatial dimensions) d - feature dimension, which is also log2(
infinity/models/bsq_vae/multiscale_bsq.py:625
Methodforward
(self, x)
infinity/models/bsq_vae/flux_vqgan.py:36
Methodforward
(self, x: Tensor)
infinity/models/bsq_vae/flux_vqgan.py:88
Methodforward
(self, x)
infinity/models/bsq_vae/flux_vqgan.py:112
Methodforward
(self, x: Tensor)
infinity/models/bsq_vae/flux_vqgan.py:139
Methodforward
(self, x: Tensor)
infinity/models/bsq_vae/flux_vqgan.py:166
Methodforward
(self, x, return_hidden=False)
infinity/models/bsq_vae/flux_vqgan.py:251
Methodforward
(self, z)
infinity/models/bsq_vae/flux_vqgan.py:358
Methodforward
(self, x)
infinity/models/bsq_vae/flux_vqgan.py:485
Methodforward
(self, x)
infinity/models/bsq_vae/conv.py:32
Functionfused_ada_layer_norm
(C: int, eps: float, x: torch.Tensor, scale: torch.Tensor, shift: torch.Tensor)
infinity/models/fused_op.py:17
Functionfused_ada_rms_norm
(C: int, eps: float, x: torch.Tensor, scale: torch.Tensor, shift: torch.Tensor)
infinity/models/fused_op.py:24
Functionfused_rms_norm
(x: torch.Tensor, weight: nn.Parameter, eps: float)
infinity/models/fused_op.py:11
Functionget_device
()
infinity/utils/dist.py:78
Methodget_different_generator_for_each_rank
(self)
infinity/utils/arg_util.py:263
Functionget_h_div_w_template2indices
(h_div_w_list, h_div_w_templates)
infinity/utils/dynamic_resolution.py:52
Methodget_output_from_indices
(self, indices)
infinity/models/bsq_vae/multiscale_bsq.py:259
Functionget_params_num
(d, w, mlp)
infinity/models/infinity.py:755
Functionget_rank_given_group
(group: tdist.ProcessGroup)
infinity/utils/dist.py:62
Functionget_rank_str_zfill
()
infinity/utils/dist.py:66
Methodget_text_embeddings
(self, texts)
infinity/models/t5.py:95
Functionget_world_size
()
infinity/utils/dist.py:74
Functionglob_with_global_step
(pattern, recursive=False)
infinity/utils/save_and_load.py:28
Methodglobal_avg
(self)
infinity/utils/misc.py:190
Methodgpt_training
(self)
infinity/utils/arg_util.py:244
Functionhandle_timeout
(signum, frame)
infinity/utils/amp_opt.py:23
Functionidentity
(t)
infinity/models/bsq_vae/multiscale_bsq.py:56
← previousnext →201–300 of 369, ranked by callers