| 18 | /// Identical architecture to DecoderBlock in vae_decoder.rs. |
| 19 | #[derive(Debug, Clone)] |
| 20 | struct EncoderBlock { |
| 21 | norm_weight: Tensor, |
| 22 | ffn_norm_weight: Tensor, |
| 23 | eps: f32, |
| 24 | gamma: Tensor, |
| 25 | mixer_weight: Tensor, |
| 26 | mixer_bias: Tensor, |
| 27 | ffn_gamma: Tensor, |
| 28 | ffn_linear1_weight: Tensor, |
| 29 | ffn_linear1_bias: Option<Tensor>, |
| 30 | ffn_linear2_weight: Tensor, |
| 31 | ffn_linear2_bias: Option<Tensor>, |
| 32 | backend: Arc<dyn ComputeBackend>, |
| 33 | } |
| 34 | |
| 35 | impl EncoderBlock { |
| 36 | fn load(vb: VarBuilder, channels: usize, eps: f64, backend: Arc<dyn ComputeBackend>) -> Result<Self> { |
nothing calls this directly
no outgoing calls
no test coverage detected