MCPcopy Index your code

hub / github.com/apple/ml-stable-diffusion / types & classes

Types & classes29 in github.com/apple/ml-stable-diffusion

⨍ Functions 150◇ Types & classes 29
↓ 6 callersClassResnetBlock2D
python_coreml_stable_diffusion/unet.py:399
↓ 5 callersClassTestStableDiffusionForTextToImage
Test Stable Diffusion text-to-image pipeline for: - PyTorch to CoreML conversion via coremltools - Speed of CoreML runtime across several co
tests/test_stable_diffusion.py:40
↓ 3 callersClassLayerNormANE
LayerNorm optimized for Apple Neural Engine (ANE) execution Note: This layer only supports normalization over the final dim. It expects `num_cha
python_coreml_stable_diffusion/layer_norm.py:11
↓ 3 callersClassSpatialTransformer
python_coreml_stable_diffusion/unet.py:506
↓ 3 callersClassTimestepEmbedding
python_coreml_stable_diffusion/unet.py:623
↓ 3 callersClassTimesteps
python_coreml_stable_diffusion/unet.py:678
↓ 2 callersClassCoreMLModel
Wrapper for running CoreML models using coremltools
python_coreml_stable_diffusion/coreml_model.py:21
↓ 2 callersClassCrossAttention
Apple Silicon friendly version of `diffusers.models.attention.CrossAttention`
python_coreml_stable_diffusion/unet.py:46
↓ 2 callersClassDownsample2D
python_coreml_stable_diffusion/unet.py:496
↓ 2 callersClassUNetMidBlock2DCrossAttn
python_coreml_stable_diffusion/unet.py:724
↓ 2 callersClassUpsample2D
python_coreml_stable_diffusion/unet.py:485
↓ 1 callersClassBasicTransformerBlock
python_coreml_stable_diffusion/unet.py:559
↓ 1 callersClassControlNetConditioningEmbedding
python_coreml_stable_diffusion/controlnet.py:15
↓ 1 callersClassCoreMLStableDiffusionPipeline
Core ML version of `diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline`
python_coreml_stable_diffusion/pipeline.py:47
↓ 1 callersClassCrossAttnDownBlock2D
python_coreml_stable_diffusion/unet.py:275
↓ 1 callersClassCrossAttnUpBlock2D
python_coreml_stable_diffusion/unet.py:144
↓ 1 callersClassDownBlock2D
python_coreml_stable_diffusion/unet.py:346
↓ 1 callersClassFeedForward
python_coreml_stable_diffusion/unet.py:587
↓ 1 callersClassGEGLU
python_coreml_stable_diffusion/unet.py:602
↓ 1 callersClassTextEncoder
python_coreml_stable_diffusion/torch2coreml.py:324
↓ 1 callersClassTextEncoderProjection
python_coreml_stable_diffusion/multilingual_projection.py:23
↓ 1 callersClassUpBlock2D
python_coreml_stable_diffusion/unet.py:221
↓ 1 callersClassVAEDecoder
Wrapper nn.Module wrapper for pipe.decode() method
python_coreml_stable_diffusion/torch2coreml.py:500
↓ 1 callersClassVAEEncoder
Wrapper nn.Module wrapper for pipe.encode() method
python_coreml_stable_diffusion/torch2coreml.py:600
ClassAttentionImplementations
python_coreml_stable_diffusion/unet.py:34
ClassControlNetModel
python_coreml_stable_diffusion/controlnet.py:49
ClassLayerNormANE
python_coreml_stable_diffusion/unet.py:134
ClassUNet2DConditionModel
python_coreml_stable_diffusion/unet.py:791
ClassUNet2DConditionModelXL
UNet2DConditionModel variant for Stable Diffusion XL with an extended forward() signature
python_coreml_stable_diffusion/unet.py:1041