MCPcopy Create free account

hub / github.com/StargazerX0/ScaleKV / functions

Functions369 in github.com/StargazerX0/ScaleKV

⨍ Functions 369◇ Types & classes 52
↓ 16 callersMethodstate_dict
(self, key_ordered=True)
infinity/utils/arg_util.py:314
↓ 14 callersFunctiongen_one_img
( infinity_test, vae, text_tokenizer, text_encoder, prompt, cfg_list=[], tau_li
tools/run_infinity.py:77
↓ 13 callersMethodupdate
(self, **kwargs)
infinity/utils/misc.py:216
↓ 9 callersMethoddecode
(self, z)
infinity/models/bsq_vae/flux_vqgan.py:532
↓ 9 callersFunctionload_tokenizer
(t5_path ='')
tools/run_infinity.py:149
↓ 9 callersFunctionload_transformer
(vae, args)
tools/run_infinity.py:279
↓ 9 callersFunctionload_visual_tokenizer
(args)
tools/run_infinity.py:257
↓ 7 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, drop=0., fused_mlp=False)
infinity/models/basic.py:143
↓ 7 callersMethod__init__
( self, ch: int, ch_mult: list[int], num_res_blocks: int, z_channels:
infinity/models/bsq_vae/flux_vqgan.py:185
↓ 7 callersFunctiondefault
(*args)
infinity/models/bsq_vae/multiscale_bsq.py:59
↓ 5 callersFunctionexists
(v)
infinity/models/bsq_vae/multiscale_bsq.py:53
↓ 4 callersMethod__init__
Build pretrained InceptionV3 Parameters ---------- output_blocks : list of int Indices of blocks to return featur
tools/inception.py:90
↓ 4 callersMethodadd_lvl_embeding_for_x_BLC
(self, x_BLC, scale_schedule, need_to_pad=0)
infinity/models/infinity.py:352
↓ 4 callersMethodflush
(self)
infinity/utils/dist.py:305
↓ 4 callersFunctionget_dropout_layer
(p)
infinity/models/basic.py:138
↓ 4 callersMethodindices_to_codes
( self, indices, label_type = 'int_label', project_out = True )
infinity/models/bsq_vae/multiscale_bsq.py:536
↓ 4 callersMethodkv_caching
(self, enable: bool)
infinity/models/basic.py:242
↓ 4 callersFunctionswish
(x: Tensor)
infinity/models/bsq_vae/flux_vqgan.py:51
↓ 3 callersMethod__init__
( self, *, dim = None, codebook_size = None, entropy_loss_weight = 0.1
infinity/models/bsq_vae/multiscale_bsq.py:395
↓ 3 callersMethodadd_lvl_embeding
(self, feature, scale_ind, scale_schedule, need_to_pad=0)
infinity/models/infinity.py:344
↓ 3 callersFunctionbarrier
()
infinity/utils/dist.py:126
↓ 3 callersFunctioncompute_statistics_of_path
(path, model, batch_size, dims, device, max_samples, num_workers=1)
tools/fid_score.py:244
↓ 3 callersMethodencode
(self, x, scale_schedule, return_residual_norm_per_scale=False)
infinity/models/bsq_vae/flux_vqgan.py:526
↓ 3 callersMethodget_entropy
(self, count, dim=-1, eps=1e-4, normalize=True)
infinity/models/bsq_vae/multiscale_bsq.py:617
↓ 3 callersMethodget_logits
:param h: hidden_state, shaped (B or batch_size, L or seq_len, C or hidden_dim) :param cond_BD: shaped (B or batch_size, D or cond_di
infinity/models/infinity.py:334
↓ 3 callersFunctionis_local_master
()
infinity/utils/dist.py:96
↓ 3 callersMethodloggable
(self)
infinity/utils/misc.py:71
↓ 3 callersMethodmax
(self)
infinity/utils/misc.py:194
↓ 3 callersFunctiontime_str
(fmt='[%m-%d %H:%M:%S]')
infinity/utils/misc.py:42
↓ 3 callersFunctiontqdm
(x)
tools/fid_score.py:49
↓ 3 callersFunctionvae_model
(vqgan_ckpt, schedule_mode, codebook_dim, codebook_size, test_mode=True, patch_size=16, encoder_ch_mult=[1, 2,
infinity/models/bsq_vae/vae.py:72
↓ 3 callersMethodwrite
(self, message)
infinity/utils/dist.py:301
↓ 2 callersMethod__init__
( self, vae_local, text_channels=0, text_maxlen=0, # text-cond generation selectin
infinity/models/infinity.py:75
↓ 2 callersFunctionadd_common_arguments
(parser)
tools/run_infinity.py:351
↓ 2 callersMethodautoregressive_infer_cfg
( self, vae=None, scale_schedule=None, label_B_or_BLT=None, B=1, negat
infinity/models/infinity.py:456
↓ 2 callersMethodbasic_clean
(text)
infinity/models/t5.py:141
↓ 2 callersFunctionbuild_2d_sincos_position_embedding
(h, w, embed_dim, temperature=10000., sc=0, verbose=True)
infinity/utils/misc.py:300
↓ 2 callersFunctioncalculate_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/fid_score.py:162
↓ 2 callersMethodclose
(self)
infinity/utils/misc.py:101
↓ 2 callersFunctionenable_scale_kv
(model, window_size=16, max_capacity=121, kernel_size=5, pooling='maxpool')
enable_scale_kv_8b.py:33
↓ 2 callersFunctionenable_scale_kv
(model, window_size=16, max_capacity=121, kernel_size=5, pooling='maxpool')
enable_scale_kv_2b.py:33
↓ 2 callersFunctionencode_prompt
(text_tokenizer, text_encoder, prompt, enable_positive_prompt=False)
tools/run_infinity.py:38
↓ 2 callersFunctionextract_ep_iter
(filename)
infinity/utils/save_and_load.py:18
↓ 2 callersMethodflip_quant
(self, x)
infinity/models/bsq_vae/multiscale_bsq.py:264
↓ 2 callersMethodflush
(self)
infinity/utils/misc.py:98
↓ 2 callersMethodget_codes_from_indices
(self, indices_list)
infinity/models/bsq_vae/multiscale_bsq.py:248
↓ 2 callersFunctionget_local_rank
()
infinity/utils/dist.py:70
↓ 2 callersFunctionget_part_jsonls
(filepath, total_line_number, parts=512)
infinity/utils/large_file_util.py:19
↓ 2 callersMethodinfinite_next
(self, generator_info)
infinity/dataset/dataset_t2i_iterable.py:276
↓ 2 callersFunctionis_main_process
()
infinity/utils/wandb_utils.py:12
↓ 2 callersFunctionis_master
()
infinity/utils/dist.py:92
↓ 2 callersFunctionl2norm
(t)
infinity/models/bsq_vae/multiscale_bsq.py:74
↓ 2 callersFunctionlabels2image
(all_indices, label_type='int_label', scale_schedule=None)
infinity/models/bitwise_self_correction.py:9
↓ 2 callersFunctionload_cnn
(model, state_dict, prefix, expand=False, use_linear=False)
infinity/models/bsq_vae/vae.py:6
↓ 2 callersMethodload_state_dict
(self, d: Union[OrderedDict, dict, str])
infinity/utils/arg_util.py:322
↓ 2 callersFunctionread_image
input: path output: tensor (C, H, W)
scripts/compute_metrics.py:14
↓ 2 callersMethodrun
(self)
infinity/utils/misc.py:133
↓ 2 callersFunctionsample_with_top_k_top_p_also_inplace_modifying_logits_
(logits_BlV: torch.Tensor, top_k: int = 0, top_p: float = 0.0, rng=None, num_samples=1)
infinity/models/infinity.py:721
↓ 2 callersMethodset_epoch
(self, epoch)
infinity/dataset/dataset_t2i_iterable.py:197
↓ 2 callersMethodset_generator
(self, )
infinity/dataset/dataset_t2i_iterable.py:201
↓ 2 callersFunctionsplit_large_txt_files
(filepath, chunk_id2save_files)
infinity/utils/large_file_util.py:37
↓ 2 callersFunctionunpack_one
(t, ps, pattern)
infinity/models/bsq_vae/multiscale_bsq.py:71
↓ 2 callersMethodupdate_kv
(self, key_states, query_states, value_states, scale_ind)
scale_kv.py:157
↓ 1 callersMethod__init__
(self, module, *args, **kwargs)
infinity/utils/misc.py:291
↓ 1 callersFunction__initialize
(fork=False, backend='nccl', gpu_id_if_not_distibuted=0, timeout_minutes=30)
infinity/utils/dist.py:23
↓ 1 callersFunction_causal_mask
(b, h, q_idx, kv_idx)
infinity/models/flex_attn.py:16
↓ 1 callersFunction_change_builtin_print
(is_master)
infinity/utils/dist.py:262
↓ 1 callersFunction_ex_repr
(self)
infinity/models/__init__.py:8
↓ 1 callersMethod_forward
(self, x: Tensor, return_hidden=False)
infinity/models/bsq_vae/flux_vqgan.py:257
↓ 1 callersMethod_forward
(self, z: Tensor)
infinity/models/bsq_vae/flux_vqgan.py:364
↓ 1 callersFunction_generate_var_infer_mask_with_kv_cache
(lengths)
infinity/models/flex_attn.py:65
↓ 1 callersFunction_generate_var_mask_mod
Generates mask mods that apply to inputs to flex attention in the sequence stacked format. Args: offsets: This tensor should be of sh
infinity/models/flex_attn.py:32
↓ 1 callersFunction_length_to_offsets
Converts a list of lengths to a list of offsets. Args: lengths: A list of lengths.
infinity/models/flex_attn.py:19
↓ 1 callersMethod_next_h_div_w_template
(self,)
infinity/dataset/dataset_t2i_iterable.py:215
↓ 1 callersFunction_offsets_to_doc_ids_tensor
(offsets)
infinity/models/flex_attn.py:49
↓ 1 callersFunction_pil_interp
(method)
tools/run_tokenizer.py:37
↓ 1 callersMethod_select_window_indices
(self, current_scale_size, current_scale_start, device, bsz, num_heads, key_len)
scale_kv.py:33
↓ 1 callersFunctionallreduce
(t: torch.Tensor, async_op=False)
infinity/utils/dist.py:131
↓ 1 callersFunctionapply_rotary_emb
(q, k, scale_schedule, rope2d_freqs_grid, pad_to_multiplier, rope2d_normalized_by_hw, scale_ind)
infinity/models/basic.py:96
↓ 1 callersFunctionarray2grid
(x)
infinity/utils/wandb_utils.py:51
↓ 1 callersMethodattention
(self, h_: Tensor)
infinity/models/bsq_vae/flux_vqgan.py:72
↓ 1 callersFunctionaug_with_positive_prompt
(prompt)
tools/run_infinity.py:59
↓ 1 callersMethodbits_to_codes
(self, bits)
infinity/models/bsq_vae/multiscale_bsq.py:529
↓ 1 callersFunctioncalculate_activation_statistics
Calculation of the statistics used by the FID. Params: -- files : List of image files paths -- model : Instance of inception m
tools/fid_score.py:219
↓ 1 callersFunctioncalculate_fid_given_paths
Calculates the FID of two paths
tools/fid_score.py:262
↓ 1 callersFunctioncleanup
()
tools/run_tokenizer.py:230
↓ 1 callersMethodclose
(self)
infinity/utils/dist.py:312
↓ 1 callersMethodcompile_flex_attn
(self)
infinity/models/infinity.py:293
↓ 1 callersMethoddecode_from_indices
(self, all_indices, scale_schedule, label_type)
infinity/models/bsq_vae/flux_vqgan.py:537
↓ 1 callersMethodencode_for_raw_features
(self, x, scale_schedule, return_residual_norm_per_scale=False)
infinity/models/bsq_vae/flux_vqgan.py:509
↓ 1 callersFunctionfid_inception_v3
Build pretrained Inception model for FID computation The Inception model for FID computation uses a different set of weights and has a slight
tools/inception.py:225
↓ 1 callersMethodforward
label_B_or_BLT: label_B or (kv_compact, cu_seqlens_k, max_seqlen_k) :return: logits BLV, V is vocab_size
infinity/models/infinity.py:366
↓ 1 callersFunctiongenerate_run_id
(exp_name)
infinity/utils/wandb_utils.py:22
↓ 1 callersFunctionget_activations
Calculates the activations of the pool_3 layer for all images. Params: -- files : List of image files paths -- model : Instan
tools/fid_score.py:95
↓ 1 callersFunctionget_args
()
tools/run_tokenizer.py:214
↓ 1 callersFunctionget_cache_sizes
()
enable_scale_kv_8b.py:9
↓ 1 callersFunctionget_cache_sizes
()
enable_scale_kv_2b.py:9
↓ 1 callersFunctionget_ema_model
(model)
infinity/models/ema.py:6
↓ 1 callersFunctionget_gpt_result
Retrieves a chat response using the GPT-4 model. Args: model_name (str, optional): The name of the GPT model to use. Defa
tools/prompt_rewriter.py:85
↓ 1 callersMethodget_h_div_w_template_2_unlearned_batches
(self,)
infinity/dataset/dataset_t2i_iterable.py:205
next →1–100 of 369, ranked by callers