MCPcopy Create free account

hub / github.com/StargazerX0/ScaleKV / functions

Functions369 in github.com/StargazerX0/ScaleKV

⨍ Functions 369◇ Types & classes 52
Functioninference_eval
(rank, world_size, args, vae, return_dict, val_txt, tgt_h, tgt_w, augmentations)
tools/run_tokenizer.py:64
Functioninfinity_20b
(depth=58, embed_dim=4608, num_heads=4608//128, drop_path_rate=0.25, **kwargs)
infinity/models/infinity.py:774
Functioninfinity_2b
(depth=32, embed_dim=2048, num_heads=2048//128, drop_path_rate=0.1, **kwargs)
infinity/models/infinity.py:771
Functioninfinity_layer12
(depth=12, embed_dim=768, num_heads=8, drop_path_rate=0.1, **kwargs)
infinity/models/infinity.py:778
Functioninfinity_layer16
(depth=16, embed_dim=1152, num_heads=12, drop_path_rate=0.1, **kwargs)
infinity/models/infinity.py:781
Functioninfinity_layer24
(depth=24, embed_dim=1536, num_heads=16, drop_path_rate=0.1, **kwargs)
infinity/models/infinity.py:784
Functioninfinity_layer32
(depth=32, embed_dim=2080, num_heads=20, drop_path_rate=0.1, **kwargs)
infinity/models/infinity.py:787
Functioninfinity_layer40
(depth=40, embed_dim=2688, num_heads=24, drop_path_rate=0.1, **kwargs)
infinity/models/infinity.py:790
Functioninfinity_layer48
(depth=48, embed_dim=3360, num_heads=28, drop_path_rate=0.1, **kwargs)
infinity/models/infinity.py:793
Functioninit_dist_and_get_args
()
infinity/utils/arg_util.py:354
Functioninit_distributed_mode
(local_out_path, fork=False, only_sync_master=False, timeout_minutes=30)
infinity/utils/dist.py:248
Functioninit_weights
:param model: the model to be inited :param conv_std_or_gain: how to init every conv layer `m` > 0: nn.init.trunc_normal_(m.weight.da
infinity/models/init_param.py:4
Functioninitialize
(args, entity, exp_name, project_name)
infinity/utils/wandb_utils.py:27
Methodis_gpt_visualization_only
(self)
infinity/utils/arg_util.py:236
Functionis_pow2n
(x)
infinity/utils/misc.py:38
Methodis_vae_visualization_only
(self)
infinity/utils/arg_util.py:232
Methodisatty
(self)
infinity/utils/dist.py:309
Functionload_csv_as_dicts
(csv_filename)
infinity/utils/csv_util.py:17
Functionload_save
(reso=512)
infinity/dataset/build.py:79
Methodload_state_dict
(self, state, strict=True)
infinity/utils/amp_opt.py:183
Methodload_state_dict
(self, state_dict: Dict[str, Any], strict=False, assign=False)
infinity/models/infinity.py:648
Functionlocal_master_only
(func)
infinity/utils/dist.py:218
Functionlog
(stats, step=None)
infinity/utils/wandb_utils.py:40
Methodlog_every
(self, start_it, max_iters, itrt, log_freq, log_every_iter=False, header='')
infinity/utils/misc.py:248
Functionlog_image
(name, sample, step=None)
infinity/utils/wandb_utils.py:45
Methodlog_image
(self, tag, img_chw, step=None)
infinity/utils/misc.py:92
Methodlog_param
(self, ep: int)
infinity/utils/amp_opt.py:100
Methodlog_tensor_as_distri
(self, tag, tensor1d, step=None)
infinity/utils/misc.py:83
Functionmaster_only
(func)
infinity/utils/dist.py:205
Functionmaybe_distributed_mean
(t)
infinity/models/bsq_vae/multiscale_bsq.py:43
Methodmedian
(self)
infinity/utils/misc.py:182
Functionnew_group
(ranks: List[int])
infinity/utils/dist.py:113
Functionnew_local_machine_group
()
infinity/utils/dist.py:119
Functionnormalize_01_into_pm1
(x)
infinity/dataset/build.py:37
Functionos_system_get_stdout
(cmd)
infinity/utils/misc.py:24
Functionos_system_get_stdout_stderr
(cmd)
infinity/utils/misc.py:26
Functionparallelize
(net, syncbn=False)
infinity/utils/dist.py:105
Functionpatched_self_attn_forward
(self, x, attn_bias_or_two_vector=None, attn_fn=None, scale_schedule=None,
enable_scale_kv_8b.py:36
Functionpatched_self_attn_forward
(self, x, attn_bias_or_two_vector=None, attn_fn=None, scale_schedule=None,
enable_scale_kv_2b.py:36
Functionper_param_clip_grad_norm_
(parameters, thresh: float, stable=False, fp=None)
infinity/utils/amp_opt.py:27
Functionpil_load
(path: str, proposal_size)
infinity/dataset/build.py:151
Functionprint_aug
(transform, label)
infinity/dataset/build.py:112
Functionprt
(*args, **kwargs)
infinity/utils/dist.py:269
Methodquantize
(self, z)
infinity/models/bsq_vae/multiscale_bsq.py:579
Functionrewrite
(im: PImage, file: str, info: str)
infinity/dataset/build.py:166
Functionround_up_multiple
(num, mult)
infinity/models/bsq_vae/multiscale_bsq.py:65
Functionsampling_with_top_k_top_p_also_inplace_modifying_probs_
(probs_BlV: torch.Tensor, top_k: int = 0, top_p: float = 0.0, rng=None, num_samples=1)
infinity/models/infinity.py:737
Methodsav
( self, args: arg_util.Args, g_it: int, next_ep: int, next_it: int, trainer, acc_str: Optional
infinity/utils/save_and_load.py:47
Functionsave_lines
(lines, filename)
infinity/utils/large_file_util.py:13
Functionset_gpu_id
(gpu_id: int)
infinity/utils/dist.py:82
Methodset_initial_seed
(self, benchmark: bool)
infinity/utils/arg_util.py:247
Methodset_step
(self, step=None)
infinity/utils/misc.py:65
Methodsoft_entropy_loss
(self, z)
infinity/models/bsq_vae/multiscale_bsq.py:599
Methodspecial_init
( self, aln_init: float, aln_gamma_init: float, scale_head: float, sca
infinity/models/infinity.py:665
Methodsplit_and_sleep
(generator_info)
infinity/dataset/dataset_t2i_iterable.py:159
Methodstate_dict
(self)
infinity/utils/amp_opt.py:175
Methodsynchronize_between_processes
Warning: does not synchronize the deque!
infinity/utils/misc.py:170
Methodsynchronize_between_processes
(self)
infinity/utils/misc.py:241
Methodtime_preds
(self, counts)
infinity/utils/misc.py:201
Functiontime_str
(fmt='[%m-%d %H:%M:%S]')
infinity/dataset/build.py:33
Methodtouch_log
(self)
infinity/utils/arg_util.py:311
Methodupdate
(self, head='scalar', step=None, **kwargs)
infinity/utils/misc.py:74
Methodupdate
(self, value, n=1)
infinity/utils/misc.py:165
Functionupdate_ema
Step the EMA model towards the current model.
infinity/models/ema.py:14
Methodvalue
(self)
infinity/utils/misc.py:198
Functionvar_mask_mod
(b, h, q_idx, kv_idx)
infinity/models/flex_attn.py:58
Methodvis_key_params
(self, ep)
infinity/models/infinity.py:645
Functionwrapper
(*args, **kwargs)
infinity/utils/dist.py:207
Functionwrite_dicts2csv_file
(input_dict_list, csv_filename)
infinity/utils/csv_util.py:8
← previous301–369 of 369, ranked by callers