MCPcopy Create free account

hub / github.com/StargazerX0/ScaleKV / types & classes

Types & classes52 in github.com/StargazerX0/ScaleKV

⨍ Functions 369◇ Types & classes 52
↓ 20 callersClassConv
infinity/models/bsq_vae/conv.py:7
↓ 9 callersClassInfinity
infinity/models/infinity.py:74
↓ 6 callersClassResnetBlock
infinity/models/bsq_vae/flux_vqgan.py:92
↓ 5 callersClassNormalize
infinity/models/bsq_vae/flux_vqgan.py:19
↓ 4 callersClassFIDInceptionC
InceptionC block patched for FID computation
tools/inception.py:278
↓ 3 callersClassCrossAttention
infinity/models/basic.py:318
↓ 3 callersClassFIDInceptionA
InceptionA block patched for FID computation
tools/inception.py:253
↓ 3 callersClassInceptionV3
Pretrained InceptionV3 network returning feature maps
tools/inception.py:75
↓ 2 callersClassAttnBlock
infinity/models/bsq_vae/flux_vqgan.py:60
↓ 2 callersClassBackupStreamToFile
infinity/utils/dist.py:287
↓ 2 callersClassFlexAttn
infinity/models/flex_attn.py:72
↓ 2 callersClassScaleKVCluster
scale_kv.py:6
↓ 2 callersClassSelfAttention
infinity/models/basic.py:195
↓ 2 callersClassSmoothedValue
Track a series of values and provide access to smoothed values over a window or the global series average.
infinity/utils/misc.py:152
↓ 2 callersClassT2IIterableDataset
infinity/dataset/dataset_t2i_iterable.py:84
↓ 1 callersClassAdaLNBeforeHead
infinity/models/basic.py:525
↓ 1 callersClassArgs
infinity/utils/arg_util.py:18
↓ 1 callersClassAutoEncoder
infinity/models/bsq_vae/flux_vqgan.py:390
↓ 1 callersClassBSQ
infinity/models/bsq_vae/multiscale_bsq.py:394
↓ 1 callersClassDecoder
infinity/models/bsq_vae/flux_vqgan.py:288
↓ 1 callersClassDownsample
infinity/models/bsq_vae/flux_vqgan.py:128
↓ 1 callersClassEncoder
infinity/models/bsq_vae/flux_vqgan.py:184
↓ 1 callersClassFIDInceptionE_1
First InceptionE block patched for FID computation
tools/inception.py:306
↓ 1 callersClassFIDInceptionE_2
Second InceptionE block patched for FID computation
tools/inception.py:339
↓ 1 callersClassFastRMSNorm
infinity/models/basic.py:119
↓ 1 callersClassImagePathDataset
tools/fid_score.py:79
↓ 1 callersClassLayerNorm
r""" LayerNorm that supports two data formats: channels_last (default) or channels_first. The ordering of the dimensions in the inputs. channels_
infinity/models/bsq_vae/multiscale_bsq.py:135
↓ 1 callersClassMultiImageDataset
scripts/compute_metrics.py:26
↓ 1 callersClassMultiInpIdentity
infinity/models/infinity.py:33
↓ 1 callersClassMultiScaleBSQ
Follows Algorithm 1. in https://arxiv.org/pdf/2107.03312.pdf
infinity/models/bsq_vae/multiscale_bsq.py:170
↓ 1 callersClassMultipleLayers
infinity/models/infinity.py:58
↓ 1 callersClassNullCtx
infinity/utils/amp_opt.py:15
↓ 1 callersClassPromptRewriter
tools/prompt_rewriter.py:69
↓ 1 callersClassSharedAdaLin
infinity/models/infinity.py:52
↓ 1 callersClassTextAttentivePool
infinity/models/infinity.py:38
↓ 1 callersClassUpsample
infinity/models/bsq_vae/flux_vqgan.py:145
ClassAmpOptimizer
infinity/utils/amp_opt.py:45
ClassBitwiseSelfCorrection
infinity/models/bitwise_self_correction.py:23
ClassCKPTSaver
infinity/utils/save_and_load.py:38
ClassCosineSimLinear
infinity/models/bsq_vae/multiscale_bsq.py:87
ClassCrossAttnBlock
infinity/models/basic.py:454
ClassDistLogger
infinity/utils/misc.py:46
ClassFFN
infinity/models/basic.py:142
ClassFFNSwiGLU
infinity/models/basic.py:176
ClassLow_GPU_usage
infinity/utils/misc.py:105
ClassMetricLogger
infinity/utils/misc.py:209
ClassNullDDP
infinity/utils/misc.py:290
ClassRandomResize
infinity/dataset/build.py:66
ClassSelfAttnBlock
infinity/models/basic.py:407
ClassT5Embedder
infinity/models/t5.py:15
ClassTensorboardLogger
infinity/utils/misc.py:57
ClassTouchingDaemonDontForgetToStartMe
infinity/utils/misc.py:118