MCPcopy Create free account

hub / github.com/NeuralCarver/Michelangelo / types & classes

Types & classes68 in github.com/NeuralCarver/Michelangelo

⨍ Functions 318◇ Types & classes 68
↓ 4 callersClassMLP
michelangelo/models/modules/transformer_blocks.py:229
↓ 4 callersClassResidualAttentionBlock
michelangelo/models/modules/transformer_blocks.py:77
↓ 3 callersClassAdaLayerNorm
michelangelo/models/modules/diffusion_transformer.py:18
↓ 3 callersClassLearnedFourierEmbedder
following @crowsonkb "s lead with learned sinusoidal pos emb
michelangelo/models/modules/embedder.py:108
↓ 2 callersClassCLIPEmbedOutput
michelangelo/models/conditional_encoders/clip.py:17
↓ 2 callersClassDiagonalGaussianDistribution
michelangelo/models/modules/distributions.py:25
↓ 2 callersClassFourierEmbedder
The sin/cosine positional embedding. Given an input tensor `x` of shape [n_batch, ..., c_dim], it converts each feature dimension of `x[..., i]` i
michelangelo/models/modules/embedder.py:11
↓ 2 callersClassLatent2MeshOutput
michelangelo/models/tsal/tsal_base.py:16
↓ 2 callersClassMLP
michelangelo/models/modules/transformer_vit.py:256
↓ 2 callersClassMultiheadAttention
michelangelo/models/modules/transformer_blocks.py:18
↓ 2 callersClassMultiheadCrossAttention
michelangelo/models/modules/transformer_blocks.py:118
↓ 2 callersClassResidualCrossAttentionBlock
michelangelo/models/modules/transformer_blocks.py:188
↓ 2 callersClassTransformer
michelangelo/models/modules/transformer_blocks.py:247
↓ 1 callersClassAlignedMeshOutput
michelangelo/models/tsal/tsal_base.py:23
↓ 1 callersClassCompose
Composes several transforms together. This transform does not support torchscript. Please, see the note below. Args: transforms (list
michelangelo/data/transforms.py:341
↓ 1 callersClassCrossAttentionDecoder
michelangelo/models/tsal/sal_perceiver.py:115
↓ 1 callersClassCrossAttentionEncoder
michelangelo/models/tsal/sal_perceiver.py:20
↓ 1 callersClassDitBlock
michelangelo/models/modules/diffusion_transformer.py:37
↓ 1 callersClassMultiheadAttention
michelangelo/models/modules/transformer_vit.py:82
↓ 1 callersClassMultiheadCrossAttention
michelangelo/models/modules/transformer_vit.py:165
↓ 1 callersClassPoint2MeshOutput
michelangelo/models/tsal/tsal_base.py:8
↓ 1 callersClassQKVMultiheadAttention
michelangelo/models/modules/transformer_blocks.py:48
↓ 1 callersClassQKVMultiheadAttention
michelangelo/models/modules/transformer_vit.py:108
↓ 1 callersClassQKVMultiheadCrossAttention
michelangelo/models/modules/transformer_blocks.py:155
↓ 1 callersClassQKVMultiheadCrossAttention
michelangelo/models/modules/transformer_vit.py:197
↓ 1 callersClassRandomResize
Apply randomly Resize with a given probability.
michelangelo/data/transforms.py:279
↓ 1 callersClassResidualAttentionBlock
michelangelo/models/modules/transformer_vit.py:128
↓ 1 callersClassUNetDiffusionTransformer
michelangelo/models/modules/diffusion_transformer.py:135
ClassASLDiffuser
michelangelo/models/asl_diffusion/asl_diffuser_pl_module.py:36
ClassAbstractDistribution
michelangelo/models/modules/distributions.py:6
ClassAbstractEncoder
michelangelo/models/conditional_encoders/encoder_factory.py:13
ClassAlignedShapeAsLatentModule
michelangelo/models/tsal/tsal_base.py:86
ClassAlignedShapeAsLatentPLModule
michelangelo/models/tsal/tsal_base.py:67
ClassAlignedShapeAsLatentPLModule
michelangelo/models/tsal/asl_pl_module.py:24
ClassAlignedShapeLatentPerceiver
michelangelo/models/tsal/sal_perceiver.py:309
ClassAxisScale
michelangelo/data/transforms.py:250
ClassAxisScaleTransform
michelangelo/data/transforms.py:201
ClassBaseDenoiser
michelangelo/models/asl_diffusion/base.py:7
ClassCLIPAlignedShapeAsLatentModule
michelangelo/models/tsal/clip_asl_module.py:11
ClassCLIPEncoder
michelangelo/models/conditional_encoders/clip.py:23
ClassCheckpointFunction
michelangelo/models/modules/checkpoint.py:38
ClassClassEmbedder
michelangelo/models/conditional_encoders/encoder_factory.py:23
ClassClipASLDiffuser
michelangelo/models/asl_diffusion/clip_asl_diffuser_pl_module.py:33
ClassConditionalASLUDTDenoiser
michelangelo/models/asl_diffusion/asl_udt.py:13
ClassContrastKLNearFar
michelangelo/models/tsal/loss.py:193
ClassDiT
michelangelo/models/modules/diffusion_transformer.py:92
ClassDiracDistribution
michelangelo/models/modules/distributions.py:14
ClassFeatureSelection
michelangelo/data/transforms.py:183
ClassFrozenAlignedCLIPTextEmbedder
Uses the CLIP transformer encoder for text (from Hugging Face)
michelangelo/models/conditional_encoders/encoder_factory.py:111
ClassFrozenCLIPImageEmbedder
Uses the CLIP transformer encoder for text (from Hugging Face)
michelangelo/models/conditional_encoders/encoder_factory.py:185
ClassFrozenCLIPImageGridEmbedder
michelangelo/models/conditional_encoders/encoder_factory.py:299
ClassFrozenCLIPTextEmbedder
Uses the CLIP transformer encoder for text (from Hugging Face)
michelangelo/models/conditional_encoders/encoder_factory.py:38
ClassKLNearFar
michelangelo/models/tsal/loss.py:13
ClassKLNearFarColor
michelangelo/models/tsal/loss.py:95
ClassMeshOutput
michelangelo/graphics/primitives/mesh.py:67
ClassMoECLIPImageEncoder
michelangelo/models/conditional_encoders/encoder_factory.py:379
ClassPyThreeJSViewer
michelangelo/utils/visualizers/pythreejs_viewer.py:12
ClassRandomSample
michelangelo/data/transforms.py:85
ClassResidualCrossAttentionBlock
michelangelo/models/modules/transformer_vit.py:219
ClassShapeAsLatentModule
michelangelo/models/tsal/tsal_base.py:51
ClassShapeAsLatentPLModule
michelangelo/models/tsal/tsal_base.py:35
ClassShapeAsLatentPLModule
michelangelo/models/tsal/sal_pl_module.py:22
ClassShapeAsLatentPerceiver
michelangelo/models/tsal/sal_perceiver.py:162
ClassSplitRandomSample
michelangelo/data/transforms.py:130
ClassTexturedShapeAsLatentModule
michelangelo/models/tsal/tsal_base.py:106
ClassToTensor
michelangelo/data/transforms.py:235
ClassTransformer
michelangelo/models/modules/transformer_vit.py:271
ClassTriplaneLearnedFourierEmbedder
michelangelo/models/modules/embedder.py:135