MCPcopy Create free account

hub / github.com/StargazerX0/ScaleKV / functions

Functions369 in github.com/StargazerX0/ScaleKV

⨍ Functions 369◇ Types & classes 52
↓ 1 callersFunctionget_latent2scale_schedule
(T: int, H: int, W: int, mode="original")
infinity/models/bsq_vae/multiscale_bsq.py:104
↓ 1 callersMethodget_layer_id_and_scale_exp
(self, para_name: str)
infinity/models/infinity.py:717
↓ 1 callersFunctionget_prompt_id
(prompt)
tools/run_infinity.py:131
↓ 1 callersFunctionget_rank
()
infinity/utils/dist.py:58
↓ 1 callersMethodget_text_input
(self, long_text_input, short_text_input, long_text_type)
infinity/dataset/dataset_t2i_iterable.py:299
↓ 1 callersFunctionglob_with_epoch_iter
(pattern, recursive=False)
infinity/utils/save_and_load.py:17
↓ 1 callersFunctionimage_eval
(pred_xs, pred_recs, lpips_alex, lpips_vgg, ssim_value, psnr_value, num_iter)
tools/run_tokenizer.py:192
↓ 1 callersFunctioninitialized
()
infinity/utils/dist.py:19
↓ 1 callersFunctionis_distributed
()
infinity/models/bsq_vae/multiscale_bsq.py:40
↓ 1 callersFunctionis_visualizer
()
infinity/utils/dist.py:100
↓ 1 callersFunctionjoint_vi_vae_encode_decode
(vae, image_path, scale_schedule, device, tgt_h, tgt_w)
tools/run_infinity.py:236
↓ 1 callersFunctionload_infinity
( rope2d_each_sa_layer, rope2d_normalized_by_hw, use_scale_schedule_embedding, pn, use
tools/run_infinity.py:159
↓ 1 callersFunctionlog
(t, eps = 1e-5)
infinity/models/bsq_vae/multiscale_bsq.py:79
↓ 1 callersFunctionlr_wd_annealing
Decay the learning rate with half-cycle cosine after warmup
infinity/utils/lr_control.py:9
↓ 1 callersFunctionmain
()
infer_2b.py:20
↓ 1 callersFunctionmain
()
infer_8b.py:20
↓ 1 callersFunctionmain
()
tools/fid_score.py:301
↓ 1 callersFunctionmain
()
infinity/models/basic.py:543
↓ 1 callersFunctionmain
()
scripts/sample_kv_8b.py:17
↓ 1 callersFunctionmain
()
scripts/sample_8b.py:15
↓ 1 callersFunctionmain
()
scripts/sample_kv_2b.py:17
↓ 1 callersFunctionmain
()
scripts/sample_2b.py:15
↓ 1 callersFunctionnamespace_to_dict
(namespace)
infinity/utils/wandb_utils.py:15
↓ 1 callersFunctionpack_one
(t, pattern)
infinity/models/bsq_vae/multiscale_bsq.py:68
↓ 1 callersFunctionplot
()
infinity/utils/lr_control.py:124
↓ 1 callersFunctionprecompute_rope2d_freqs_grid
(dim, dynamic_resolution_h_w, rope2d_normalized_by_hw, pad_to_multiplier=1, max_height=2048 // 16, max_width=2
infinity/models/basic.py:38
↓ 1 callersMethodprepare_model_input
(self, data_item)
infinity/dataset/dataset_t2i_iterable.py:324
↓ 1 callersFunctionprocess_short_text
(short_text)
tools/comprehensive_infer.py:11
↓ 1 callersFunctionprocess_short_text
(short_text)
infinity/dataset/dataset_t2i_iterable.py:76
↓ 1 callersMethodquantize_new
(self, z)
infinity/models/bsq_vae/multiscale_bsq.py:587
↓ 1 callersMethodrewrite
(self, prompt)
tools/prompt_rewriter.py:78
↓ 1 callersFunctionrms_norm_impl
(x, weight, epsilon)
infinity/models/basic.py:34
↓ 1 callersFunctionsave_fid_stats
Calculates the FID of two paths
tools/fid_score.py:281
↓ 1 callersFunctionsave_slim_model
(infinity_model_path, save_file=None, device='cpu', key='gpt_fsdp')
tools/run_infinity.py:137
↓ 1 callersMethodset_global_worker_id
(self)
infinity/dataset/dataset_t2i_iterable.py:184
↓ 1 callersMethodset_h_div_w_template2generator
(self,)
infinity/dataset/dataset_t2i_iterable.py:131
↓ 1 callersMethodset_tf32
(tf32: bool)
infinity/utils/arg_util.py:335
↓ 1 callersFunctionsetup
(rank, world_size)
tools/run_tokenizer.py:225
↓ 1 callersMethodsplit_meta_files
(self, )
infinity/dataset/dataset_t2i_iterable.py:156
↓ 1 callersMethodtext_preprocessing
(self, text)
infinity/models/t5.py:126
↓ 1 callersMethodtotal_samples
(self)
infinity/dataset/dataset_t2i_iterable.py:296
↓ 1 callersFunctiontransform
(pil_img, tgt_h, tgt_w)
tools/run_infinity.py:220
↓ 1 callersFunctiontransform
(pil_img, tgt_h, tgt_w)
infinity/dataset/dataset_t2i_iterable.py:59
↓ 1 callersFunctionvae_encode_decode_norm
(vae, image_path, tgt_h, tgt_w, device, augmentations)
tools/run_tokenizer.py:48
↓ 1 callersMethodvisualize
(self, vae_scale_schedule, inp_B3HW, gt_all_bit_indices, pred_all_bit_indices)
infinity/models/bitwise_self_correction.py:86
Method__call__
(self, img)
infinity/dataset/build.py:72
Method__del__
(self)
infinity/utils/dist.py:325
Method__enter__
(self)
infinity/utils/misc.py:112
Method__enter__
(self)
infinity/utils/amp_opt.py:16
Method__exit__
(self, exc_type, exc_val, exc_tb)
infinity/utils/misc.py:115
Method__exit__
(self, exc_type, exc_val, exc_tb)
infinity/utils/amp_opt.py:19
Method__getattr__
(self, attr: str)
infinity/utils/misc.py:54
Method__getattr__
(self, attr)
infinity/utils/misc.py:224
Method__getitem__
(self, i)
tools/fid_score.py:87
Method__getitem__
(self, idx)
scripts/compute_metrics.py:42
Method__init__
(self, cache_sizes, window_size=16, base_capacity=121, kernel_size=5, pooling='maxpool', layer_idx=0)
scale_kv.py:7
Method__init__
(self, files, transforms=None)
tools/fid_score.py:80
Method__init__
(self, system, few_shot_history)
tools/prompt_rewriter.py:70
Method__init__
(self, in_channels, pool_features)
tools/inception.py:255
Method__init__
(self, in_channels, channels_7x7)
tools/inception.py:280
Method__init__
(self, in_channels)
tools/inception.py:308
Method__init__
(self, in_channels)
tools/inception.py:341
Method__init__
(self, lg)
infinity/utils/misc.py:47
Method__init__
(self, log_dir, filename_suffix)
infinity/utils/misc.py:58
Method__init__
(self, files, sleep_secs, verbose)
infinity/utils/misc.py:106
Method__init__
(self, files: List[str], sleep_secs: int, verbose=False)
infinity/utils/misc.py:119
Method__init__
(self, window_size=30, fmt=None)
infinity/utils/misc.py:157
Method__init__
(self)
infinity/utils/misc.py:210
Method__init__
(self, is_master: bool, eval_milestone: List[Tuple[float, float]])
infinity/utils/save_and_load.py:39
Method__init__
( self, model_name_3letters: str, mixed_precision: int, optimizer: torch.optim.Optimiz
infinity/utils/amp_opt.py:46
Method__init__
(self, local_output_dir, for_stdout=True)
infinity/utils/dist.py:288
Method__init__
( self, meta_folder: str, max_caption_len=512, short_prob=0.2, loa
infinity/dataset/dataset_t2i_iterable.py:85
Method__init__
(self, mid_reso, final_reso, interpolation)
infinity/dataset/build.py:67
Method__init__
:param block_scales: accept VAR's block sizes like [(1,1), (2,2), (3,3)] :param mask_type: var/causal :param B: batch size
infinity/models/flex_attn.py:73
Method__init__
(self, Ct5: int, D: int)
infinity/models/infinity.py:39
Method__init__
(self, ls, num_blocks_in_a_chunk, index)
infinity/models/infinity.py:59
Method__init__
(self, device, dir_or_name='t5-v1_1-xxl', *, local_cache=False, cache_dir=None, hf_token=None, use_text_prepro
infinity/models/t5.py:20
Method__init__
(self, C, eps=1e-6, elementwise_affine=True)
infinity/models/basic.py:120
Method__init__
(self, in_features, hidden_features, out_features=None, drop=0., fused_mlp=False)
infinity/models/basic.py:177
Method__init__
:param embed_dim: model's width :param num_heads: num heads of multi-head attention :param proj_drop: always 0 for testing
infinity/models/basic.py:196
Method__init__
:param for_attn_pool: only used in VAR.text_proj_for_sos :param embed_dim: Q's dim :param kv_dim: K's and V's dim :pa
infinity/models/basic.py:319
Method__init__
( self, embed_dim, kv_dim, cross_attn_layer_scale, cond_dim, act: bool, shared_aln: bool, norm_layer:
infinity/models/basic.py:408
Method__init__
( self, embed_dim, kv_dim, cross_attn_layer_scale, cond_dim, act: bool, shared_aln: bool, norm
infinity/models/basic.py:455
Method__init__
(self, C, D, act: bool, norm_layer: partial, fused_norm_func=None)
infinity/models/basic.py:526
Method__init__
(self, vae, args)
infinity/models/bitwise_self_correction.py:24
Method__init__
( self, dim_in, dim_out, scale = 1. )
infinity/models/bsq_vae/multiscale_bsq.py:88
Method__init__
(self, normalized_shape, norm_weight=False, eps=1e-6, data_format="channels_first")
infinity/models/bsq_vae/multiscale_bsq.py:142
Method__init__
( self, *, dim, codebook_size, soft_clamp_input_value = None,
infinity/models/bsq_vae/multiscale_bsq.py:173
Method__init__
(self, in_channels, norm_type, norm_axis="spatial")
infinity/models/bsq_vae/flux_vqgan.py:20
Method__init__
(self, in_channels, norm_type='group', cnn_param=None)
infinity/models/bsq_vae/flux_vqgan.py:61
Method__init__
(self, in_channels: int, out_channels: int, norm_type='group', cnn_param=None)
infinity/models/bsq_vae/flux_vqgan.py:93
Method__init__
(self, in_channels, cnn_type="2d", spatial_down=False, temporal_down=False)
infinity/models/bsq_vae/flux_vqgan.py:129
Method__init__
(self, in_channels, cnn_type="2d", spatial_up=False, temporal_up=False, use_pxsl=False)
infinity/models/bsq_vae/flux_vqgan.py:146
Method__init__
( self, ch: int, ch_mult: list[int], num_res_blocks: int, z_channels:
infinity/models/bsq_vae/flux_vqgan.py:289
Method__init__
(self, args)
infinity/models/bsq_vae/flux_vqgan.py:391
Method__init__
(self, in_channels, out_channels, kernel_size, stride=1, padding=0, cnn_type="2d", causal_offset=0, temporal_d
infinity/models/bsq_vae/conv.py:8
Method__init__
(self, root0, root1, is_gt=False)
scripts/compute_metrics.py:27
Method__iter__
(self)
infinity/dataset/dataset_t2i_iterable.py:232
Method__len__
(self)
tools/fid_score.py:84
Method__len__
(self)
infinity/dataset/dataset_t2i_iterable.py:293
← previousnext →101–200 of 369, ranked by callers