Struct candle_transformers::models::stable_diffusion::vae::AutoEncoderKL
source · pub struct AutoEncoderKL {
pub config: AutoEncoderKLConfig,
/* private fields */
}
Fields§
§config: AutoEncoderKLConfig
Implementations§
source§impl AutoEncoderKL
impl AutoEncoderKL
pub fn new( vs: VarBuilder<'_>, in_channels: usize, out_channels: usize, config: AutoEncoderKLConfig ) -> Result<Self>
sourcepub fn encode(&self, xs: &Tensor) -> Result<DiagonalGaussianDistribution>
pub fn encode(&self, xs: &Tensor) -> Result<DiagonalGaussianDistribution>
Returns the distribution in the latent space.
Trait Implementations§
Auto Trait Implementations§
impl Freeze for AutoEncoderKL
impl !RefUnwindSafe for AutoEncoderKL
impl Send for AutoEncoderKL
impl Sync for AutoEncoderKL
impl Unpin for AutoEncoderKL
impl !UnwindSafe for AutoEncoderKL
Blanket Implementations§
source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more