add network def
Browse filesSigned-off-by: Can-Zhao <[email protected]>
configs/config_network_ddpm.json
CHANGED
|
@@ -63,7 +63,9 @@
|
|
| 63 |
"use_flash_attention": true,
|
| 64 |
"include_top_region_index_input": "@include_body_region",
|
| 65 |
"include_bottom_region_index_input": "@include_body_region",
|
| 66 |
-
"include_spacing_input": true
|
|
|
|
|
|
|
| 67 |
},
|
| 68 |
"controlnet_def": {
|
| 69 |
"_target_": "monai.apps.generation.maisi.networks.controlnet_maisi.ControlNetMaisi",
|
|
@@ -90,7 +92,9 @@
|
|
| 90 |
"num_res_blocks": 2,
|
| 91 |
"use_flash_attention": true,
|
| 92 |
"conditioning_embedding_in_channels": 8,
|
| 93 |
-
"conditioning_embedding_num_channels": [8, 32, 64]
|
|
|
|
|
|
|
| 94 |
},
|
| 95 |
"mask_generation_autoencoder_def": {
|
| 96 |
"_target_": "monai.apps.generation.maisi.networks.autoencoderkl_maisi.AutoencoderKlMaisi",
|
|
|
|
| 63 |
"use_flash_attention": true,
|
| 64 |
"include_top_region_index_input": "@include_body_region",
|
| 65 |
"include_bottom_region_index_input": "@include_body_region",
|
| 66 |
+
"include_spacing_input": true,
|
| 67 |
+
"include_fc": true,
|
| 68 |
+
"resblock_updown": false
|
| 69 |
},
|
| 70 |
"controlnet_def": {
|
| 71 |
"_target_": "monai.apps.generation.maisi.networks.controlnet_maisi.ControlNetMaisi",
|
|
|
|
| 92 |
"num_res_blocks": 2,
|
| 93 |
"use_flash_attention": true,
|
| 94 |
"conditioning_embedding_in_channels": 8,
|
| 95 |
+
"conditioning_embedding_num_channels": [8, 32, 64],
|
| 96 |
+
"include_fc": true,
|
| 97 |
+
"resblock_updown": false
|
| 98 |
},
|
| 99 |
"mask_generation_autoencoder_def": {
|
| 100 |
"_target_": "monai.apps.generation.maisi.networks.autoencoderkl_maisi.AutoencoderKlMaisi",
|