| 599 | |
| 600 | #[derive(Debug, Clone)] |
| 601 | pub struct DoubleStreamBlock { |
| 602 | img_mod: Modulation2, |
| 603 | img_norm1_weight: Tensor, |
| 604 | img_attn: SelfAttention, |
| 605 | img_norm2_weight: Tensor, |
| 606 | img_mlp: Mlp, |
| 607 | txt_mod: Modulation2, |
| 608 | txt_norm1_weight: Tensor, |
| 609 | txt_attn: SelfAttention, |
| 610 | txt_norm2_weight: Tensor, |
| 611 | txt_mlp: Mlp, |
| 612 | layer_norm_eps: f32, |
| 613 | backend: Arc<dyn ComputeBackend>, |
| 614 | } |
| 615 | |
| 616 | impl DoubleStreamBlock { |
| 617 | fn new(cfg: &Config, vb: VarBuilder, backend: Arc<dyn ComputeBackend>) -> Result<Self> { |
nothing calls this directly
no outgoing calls
no test coverage detected