Types & classes — github.com/LaurentMazare/diffusers-rs

CodeHub MCPcopy Create free account

hub / github.com/LaurentMazare/diffusers-rs / types & classes

Types & classes91 in github.com/LaurentMazare/diffusers-rs

⨍ Functions 171◇ Types & classes 91

src/transformers/clip.rs:12

examples/stable-diffusion-img2img/main.rs:20

examples/stable-diffusion/main.rs:68

examples/controlnet/main.rs:13

examples/stable-diffusion-inpaint/main.rs:26

ClassAttentionBlock

src/models/attention.rs:328

ClassAttentionBlockConfig

src/models/attention.rs:314

ClassAutoEncoderKL

src/models/vae.rs:238

ClassAutoEncoderKLConfig

src/models/vae.rs:194

ClassBasicTransformerBlock

src/models/attention.rs:162

EnumBetaSchedule

src/schedulers/mod.rs:23

ClassBlockConfig

src/models/unet_2d.rs:10

ClassClipAttention

src/transformers/clip.rs:533

ClassClipEncoder

src/transformers/clip.rs:641

ClassClipEncoderLayer

src/transformers/clip.rs:609

src/transformers/clip.rs:587

ClassClipTextEmbeddings

src/transformers/clip.rs:501

ClassClipTextTransformer

src/transformers/clip.rs:667

src/transformers/clip.rs:27

ClassControlNet

src/models/controlnet.rs:93

ClassControlNetConditioningEmbedding

src/models/controlnet.rs:8

ClassControlNetConfig

src/models/controlnet.rs:52

EnumControlType

examples/controlnet/main.rs:119

ClassCrossAttention

src/models/attention.rs:51

ClassCrossAttnDownBlock2D

src/models/unet_2d_blocks.rs:509

ClassCrossAttnDownBlock2DConfig

src/models/unet_2d_blocks.rs:487

ClassCrossAttnUpBlock2D

src/models/unet_2d_blocks.rs:685

ClassCrossAttnUpBlock2DConfig

src/models/unet_2d_blocks.rs:663

ClassDDIMScheduler

src/schedulers/ddim.rs:50

ClassDDIMSchedulerConfig

src/schedulers/ddim.rs:15

ClassDDPMScheduler

src/schedulers/ddpm.rs:51

ClassDDPMSchedulerConfig

src/schedulers/ddpm.rs:20

EnumDDPMVarianceType

src/schedulers/ddpm.rs:5

EnumDPMSolverAlgorithmType

src/schedulers/dpmsolver_multistep.rs:8

ClassDPMSolverMultistepScheduler

src/schedulers/dpmsolver_multistep.rs:80

ClassDPMSolverMultistepSchedulerConfig

src/schedulers/dpmsolver_multistep.rs:27

EnumDPMSolverType

src/schedulers/dpmsolver_multistep.rs:20

src/models/vae.rs:120

ClassDecoderConfig

src/models/vae.rs:106

ClassDeviceSetup

src/utils.rs:13

ClassDiagonalGaussianDistribution

src/models/vae.rs:212

ClassDownBlock2D

src/models/unet_2d_blocks.rs:424

ClassDownBlock2DConfig

src/models/unet_2d_blocks.rs:399

ClassDownEncoderBlock2D

src/models/unet_2d_blocks.rs:103

ClassDownEncoderBlock2DConfig

src/models/unet_2d_blocks.rs:80

ClassDownsample2D

src/models/unet_2d_blocks.rs:10

src/models/vae.rs:33

ClassEncoderConfig

src/models/vae.rs:13

ClassEulerAncestralDiscreteScheduler

src/schedulers/euler_ancestral_discrete.rs:35

ClassEulerAncestralDiscreteSchedulerConfig

src/schedulers/euler_ancestral_discrete.rs:5

ClassEulerDiscreteScheduler

src/schedulers/euler_discrete.rs:35

ClassEulerDiscreteSchedulerConfig

src/schedulers/euler_discrete.rs:5

ClassFeedForward

src/models/attention.rs:24

src/models/attention.rs:4

ClassHeunDiscreteScheduler

src/schedulers/heun_discrete.rs:30

ClassHeunDiscreteSchedulerConfig

src/schedulers/heun_discrete.rs:5

ClassKDPM2AncestralDiscreteScheduler

Scheduler created by @crowsonkb in [k_diffusion](https://github.com/crowsonkb/k-diffusion), see: https://github.com/crowsonkb/k-diffusion/blob/5b3af03

src/schedulers/k_dpm_2_ancestral_discrete.rs:34

ClassKDPM2AncestralDiscreteSchedulerConfig

src/schedulers/k_dpm_2_ancestral_discrete.rs:5

ClassKDPM2DiscreteScheduler

Scheduler created by @crowsonkb in [k_diffusion](https://github.com/crowsonkb/k-diffusion), see: https://github.com/crowsonkb/k-diffusion/blob/5b3af03

src/schedulers/k_dpm_2_discrete.rs:34

ClassKDPM2DiscreteSchedulerConfig

src/schedulers/k_dpm_2_discrete.rs:5

ClassLMSDiscreteScheduler

src/schedulers/lms_discrete.rs:35

ClassLMSDiscreteSchedulerConfig

src/schedulers/lms_discrete.rs:6

src/schedulers/integrate.rs:11

ClassPNDMScheduler

Pseudo numerical methods for diffusion models (PNDM) proposes using more advanced ODE integration techniques, namely Runge-Kutta method and a linear m

src/schedulers/pndm.rs:41

ClassPNDMSchedulerConfig

src/schedulers/pndm.rs:5

EnumPredictionType

src/schedulers/mod.rs:33

src/models/attention.rs:226

ClassResnetBlock2D

src/models/resnet.rs:44

ClassResnetBlock2DConfig

src/models/resnet.rs:11

ClassSpatialTransformer

src/models/attention.rs:233

ClassSpatialTransformerConfig

src/models/attention.rs:205

ClassStableDiffusionConfig

src/pipelines/stable_diffusion.rs:8

EnumStableDiffusionVersion

examples/stable-diffusion-img2img/main.rs:85

EnumStableDiffusionVersion

examples/stable-diffusion/main.rs:138

EnumStableDiffusionVersion

examples/stable-diffusion-inpaint/main.rs:94

ClassTimestepEmbedding

src/models/embeddings.rs:4

src/models/embeddings.rs:26

This is mostly a Rust rewrite of the original Python CLIP code. https://github.com/openai/CLIP/blob/main/clip/simple_tokenizer.py A tokenizer for CLIP

src/transformers/clip.rs:339

ClassUNet2DConditionModel

src/models/unet_2d.rs:69

ClassUNet2DConditionModelConfig

src/models/unet_2d.rs:17

EnumUNetDownBlock

src/models/unet_2d.rs:57

ClassUNetMidBlock2D

src/models/unet_2d_blocks.rs:260

ClassUNetMidBlock2DConfig

src/models/unet_2d_blocks.rs:238

ClassUNetMidBlock2DCrossAttn

src/models/unet_2d_blocks.rs:337

ClassUNetMidBlock2DCrossAttnConfig

src/models/unet_2d_blocks.rs:309

EnumUNetUpBlock

src/models/unet_2d.rs:63

src/models/unet_2d_blocks.rs:602

ClassUpBlock2DConfig

src/models/unet_2d_blocks.rs:579

ClassUpDecoderBlock2D

src/models/unet_2d_blocks.rs:184

ClassUpDecoderBlock2DConfig

src/models/unet_2d_blocks.rs:163

ClassUpsample2D

src/models/unet_2d_blocks.rs:51