Trait WithForward 

pub trait WithForward {
    // Required method
    fn forward(
        &self,
        img: &Tensor,
        img_ids: &Tensor,
        txt: &Tensor,
        txt_ids: &Tensor,
        timesteps: &Tensor,
        y: &Tensor,
        guidance: Option<&Tensor>,
    ) -> Result<Tensor>;
}

Required Methods

fn forward( &self, img: &Tensor, img_ids: &Tensor, txt: &Tensor, txt_ids: &Tensor, timesteps: &Tensor, y: &Tensor, guidance: Option<&Tensor>, ) -> Result<Tensor>

Implementors

impl WithForward for candle_transformers::models::flux::model::Flux

impl WithForward for candle_transformers::models::flux::quantized_model::Flux