Functions — github.com/LaurentMazare/diffusers-rs

hub / github.com/LaurentMazare/diffusers-rs / functions

Functions171 in github.com/LaurentMazare/diffusers-rs

⨍ Functions 171◇ Types & classes 91

src/transformers/clip.rs:400

Methodinit_noise_sigma

(&self)

src/schedulers/k_dpm_2_ancestral_discrete.rs:326

Methodinit_noise_sigma

(&self)

src/schedulers/k_dpm_2_discrete.rs:277

Methodinit_noise_sigma

(&self)

src/schedulers/lms_discrete.rs:169

Methodinit_noise_sigma

(&self)

src/schedulers/euler_ancestral_discrete.rs:135

Methodinit_noise_sigma

(&self)

src/schedulers/ddpm.rs:187

Methodinit_noise_sigma

(&self)

src/schedulers/euler_discrete.rs:143

Methodinit_noise_sigma

(&self)

src/schedulers/pndm.rs:225

Methodinit_noise_sigma

(&self)

src/schedulers/dpmsolver_multistep.rs:405

Methodinit_noise_sigma

(&self)

src/schedulers/heun_discrete.rs:210

examples/tensor-tools.rs:13

Functionmain

()

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

Functionmain

()

examples/stable-diffusion/main.rs:309

Functionmain

()

examples/controlnet/main.rs:264

Functionmain

()

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

(vs: nn::Path, c: &Config)

src/transformers/clip.rs:508

Methodnew

(inference_steps: usize, config: KDPM2AncestralDiscreteSchedulerConfig)

src/schedulers/k_dpm_2_ancestral_discrete.rs:46

Methodnew

Creates a new DDIM scheduler given the number of steps to be used for inference as well as the number of steps that was used during training.

src/schedulers/ddim.rs:63

Methodnew

(inference_steps: usize, config: KDPM2DiscreteSchedulerConfig)

src/schedulers/k_dpm_2_discrete.rs:44

Methodnew

(inference_steps: usize, config: LMSDiscreteSchedulerConfig)

src/schedulers/lms_discrete.rs:44

Methodnew

(inference_steps: usize, config: EulerAncestralDiscreteSchedulerConfig)

src/schedulers/euler_ancestral_discrete.rs:43

Methodnew

(inference_steps: usize, config: DDPMSchedulerConfig)

src/schedulers/ddpm.rs:60

Methodnew

(inference_steps: usize, config: EulerDiscreteSchedulerConfig)

src/schedulers/euler_discrete.rs:43

Methodnew

(inference_steps: usize, config: PNDMSchedulerConfig)

src/schedulers/pndm.rs:54

Methodnew

(inference_steps: usize, config: DPMSolverMultistepSchedulerConfig)

src/schedulers/dpmsolver_multistep.rs:93

Methodnew

(inference_steps: usize, config: HeunDiscreteSchedulerConfig)

src/schedulers/heun_discrete.rs:41

Methodnew

(vs: nn::Path, in_channels: i64, config: ResnetBlock2DConfig)

src/models/resnet.rs:55

Methodnew

act_fn: "silu"

src/models/embeddings.rs:11

Methodnew

( vs: nn::Path, in_channels: i64, use_conv: bool, out_channels: i64, p

src/models/unet_2d_blocks.rs:16

Methodnew

( vs: nn::Path, in_channels: i64, out_channels: i64, config: UNet2DConditionMo

src/models/unet_2d.rs:82

Methodnew

( vs: nn::Path, conditioning_embedding_channels: i64, conditioning_channels: i64,

src/models/controlnet.rs:15

Methodnew

(vs: nn::Path, dim_in: i64, dim_out: i64)

src/models/attention.rs:9

Methodnew

(vs: nn::Path, in_channels: i64, out_channels: i64, config: EncoderConfig)

src/models/vae.rs:44

Methodscale_model_input

Scales model input by (sigma^2 + 1) ^ .5

src/schedulers/k_dpm_2_ancestral_discrete.rs:240

Methodscale_model_input

Scales model input by (sigma^2 + 1) ^ .5

src/schedulers/k_dpm_2_discrete.rs:197

Methodscale_model_input

Scales the denoising model input by `(sigma^2 + 1)^0.5` to match the K-LMS algorithm.

src/schedulers/lms_discrete.rs:99

Methodscale_model_input

(&self, sample: Tensor, timestep: f64)

src/schedulers/euler_ancestral_discrete.rs:96

Methodscale_model_input

Ensures interchangeability with schedulers that need to scale the denoising model input depending on the current timestep.

src/schedulers/ddpm.rs:125

Methodscale_model_input

(&self, sample: Tensor, timestep: f64)

src/schedulers/euler_discrete.rs:96

Methodscale_model_input

Ensures interchangeability with schedulers that need to scale the denoising model input depending on the current timestep.

src/schedulers/pndm.rs:112

Methodscale_model_input

Ensures interchangeability with schedulers that need to scale the denoising model input depending on the current timestep.

src/schedulers/dpmsolver_multistep.rs:344

Methodscale_model_input

Ensures interchangeability with schedulers that need to scale the denoising model input depending on the current timestep.

src/schedulers/heun_discrete.rs:137

Methodsigma_to_t

(sigma: &Tensor, log_sigmas: Tensor)

src/schedulers/k_dpm_2_ancestral_discrete.rs:191

Methodsigma_to_t

(sigma: &Tensor, log_sigmas: Tensor)

src/schedulers/k_dpm_2_discrete.rs:148

Methodstep

(&mut self, model_output: &Tensor, timestep: f64, sample: &Tensor)

src/schedulers/k_dpm_2_ancestral_discrete.rs:258

Methodstep

(&mut self, model_output: &Tensor, timestep: f64, sample: &Tensor)

src/schedulers/k_dpm_2_discrete.rs:213

Methodstep

(&mut self, model_output: &Tensor, timestep: f64, sample: &Tensor)

src/schedulers/lms_discrete.rs:131

Methodstep

(&self, model_output: &Tensor, timestep: f64, sample: &Tensor)

src/schedulers/euler_ancestral_discrete.rs:104

Methodstep

(&self, model_output: &Tensor, timestep: usize, sample: &Tensor)

src/schedulers/ddpm.rs:129

Methodstep

(&self, model_output: &Tensor, timestep: f64, sample: &Tensor)

src/schedulers/euler_discrete.rs:104

Methodstep

(&mut self, model_output: &Tensor, timestep: usize, sample: &Tensor)

src/schedulers/pndm.rs:116

Methodstep

(&mut self, model_output: &Tensor, timestep: usize, sample: &Tensor)

src/schedulers/dpmsolver_multistep.rs:348

Methodstep

(&mut self, model_output: &Tensor, timestep: f64, sample: &Tensor)

src/schedulers/heun_discrete.rs:149

Methodtimesteps

(&self)

src/schedulers/k_dpm_2_ancestral_discrete.rs:219

Methodtimesteps

(&self)

src/schedulers/k_dpm_2_discrete.rs:176

Methodtimesteps

(&self)

src/schedulers/lms_discrete.rs:94

Methodtimesteps

(&self)

src/schedulers/euler_ancestral_discrete.rs:92

Methodtimesteps

(&self)

src/schedulers/ddpm.rs:119

Methodtimesteps

(&self)

src/schedulers/euler_discrete.rs:92

Methodtimesteps

(&self)

src/schedulers/pndm.rs:106

Methodtimesteps

(&self)

src/schedulers/dpmsolver_multistep.rs:338

Methodtimesteps

(&self)

src/schedulers/heun_discrete.rs:115

Methodv1_5

The config details can be found in the "text_config" section of this json file: https://huggingface.co/openai/clip-vit-large-patch14/blob/main/config.

src/transformers/clip.rs:44

Methodv1_5

( sliced_attention_size: Option<i64>, height: Option<i64>, width: Option<i64>, )

src/pipelines/stable_diffusion.rs:18

Methodv2_1

https://huggingface.co/stabilityai/stable-diffusion-2-1/blob/main/text_encoder/config.json

src/transformers/clip.rs:59

Methodv2_1

( sliced_attention_size: Option<i64>, height: Option<i64>, width: Option<i64>, )

src/pipelines/stable_diffusion.rs:130

Methodv2_1_

( sliced_attention_size: Option<i64>, height: Option<i64>, width: Option<i64>,

src/pipelines/stable_diffusion.rs:73

Methodv2_1_inpaint

( sliced_attention_size: Option<i64>, height: Option<i64>, width: Option<i64>, )

src/pipelines/stable_diffusion.rs:139

← previous101–171 of 171, ranked by callers