| 292 | source, |
| 293 | "flow_matching.zero_cond_embedding1", |
| 294 | {config.dim}); |
| 295 | weights.zero_cond_embedding_host = source.require_f32_tensor("flow_matching.zero_cond_embedding1", {config.dim}); |
| 296 | |
| 297 | auto & estimator = weights.estimator; |
| 298 | estimator.proj_in = load_project_layer( |
| 299 | store, |
| 300 | source, |
| 301 | "flow_matching.estimator.proj_in", |
| 302 | storage_type, |
| 303 | config.in_channels, |
| 304 | estimator_dim); |
| 305 | estimator.transformer_blocks = load_transformer_blocks( |
| 306 | store, |
| 307 | source, |
| 308 | "flow_matching.estimator.transformer_blocks", |
| 309 | storage_type, |
| 310 | config.num_layers, |
| 311 | estimator_dim); |
| 312 | estimator.connection_proj = load_project_layer( |
| 313 | store, |
| 314 | source, |
| 315 | "flow_matching.estimator.connection_proj", |
| 316 | storage_type, |
| 317 | config.in_channels + estimator_dim, |
| 318 | estimator_dim_2); |
| 319 | estimator.transformer_blocks_2 = load_transformer_blocks( |
| 320 | store, |
| 321 | source, |
| 322 | "flow_matching.estimator.transformer_blocks_2", |
| 323 | storage_type, |
| 324 | config.num_layers_2, |
| 325 | estimator_dim_2); |
| 326 | estimator.scale_shift_table = store.load_f32_tensor( |
| 327 | source, |
| 328 | "flow_matching.estimator.scale_shift_table", |
| 329 | {2, estimator_dim}); |
| 330 | estimator.scale_shift_table_2 = store.load_f32_tensor( |
| 331 | source, |
| 332 | "flow_matching.estimator.scale_shift_table_2", |
| 333 | {2, estimator_dim_2}); |
| 334 | estimator.proj_out = load_project_layer( |
| 335 | store, |
| 336 | source, |
| 337 | "flow_matching.estimator.proj_out", |
| 338 | storage_type, |
| 339 | estimator_dim_2, |
| 340 | config.out_channels); |
| 341 | estimator.adaln_single = load_adaln( |
| 342 | store, |
| 343 | source, |
| 344 | "flow_matching.estimator.adaln_single", |
| 345 | storage_type, |
| 346 | estimator_dim); |
| 347 | estimator.adaln_single_2 = load_adaln( |
| 348 | store, |
| 349 | source, |
| 350 | "flow_matching.estimator.adaln_single_2", |
| 351 | storage_type, |
no test coverage detected