| 290 | assets::require_tensor_shape(weights, "model.prediction_head.t_embedder.mlp.0.weight", {decoder.hidden_size, 256}); |
| 291 | assets::require_tensor_shape( |
| 292 | weights, |
| 293 | "model.prediction_head.layers.0.ffn.gate_proj.weight", |
| 294 | {static_cast<int64_t>(decoder.hidden_size * config.diffusion_head.head_ffn_ratio), decoder.hidden_size}); |
| 295 | assets::require_tensor_shape( |
| 296 | weights, |
| 297 | "model.prediction_head.final_layer.linear.weight", |
| 298 | {config.diffusion_head.latent_size, decoder.hidden_size}); |
| 299 |
no outgoing calls
no test coverage detected