| 226 | |
| 227 | #[derive(Debug)] |
| 228 | struct Flux2Decoder { |
| 229 | conv_in: RawConv2d, |
| 230 | mid_block: MidBlock, |
| 231 | up_blocks: Vec<UpDecoderBlock>, |
| 232 | conv_norm_out_weight: Tensor, |
| 233 | conv_norm_out_bias: Tensor, |
| 234 | conv_norm_out_groups: usize, |
| 235 | conv_norm_out_eps: f32, |
| 236 | conv_out: RawConv2d, |
| 237 | backend: Arc<dyn ComputeBackend>, |
| 238 | } |
| 239 | |
| 240 | impl Flux2Decoder { |
| 241 | fn load(vb: VarBuilder, backend: Arc<dyn ComputeBackend>) -> Result<Self> { |
nothing calls this directly
no outgoing calls
no test coverage detected