Update direct3d-s2-v-1-1/config.yaml
Browse files
direct3d-s2-v-1-1/config.yaml
CHANGED
|
@@ -158,4 +158,14 @@ refiner:
|
|
| 158 |
layers_per_block: 2
|
| 159 |
layers_mid_block: 2
|
| 160 |
patch_size: 192
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 161 |
use_fp16: true
|
|
|
|
| 158 |
layers_per_block: 2
|
| 159 |
layers_mid_block: 2
|
| 160 |
patch_size: 192
|
| 161 |
+
use_fp16: true
|
| 162 |
+
|
| 163 |
+
refiner_1024:
|
| 164 |
+
target: direct3d_s2.models.refiner.unet_refiner.Voxel_RefinerXL_sign
|
| 165 |
+
params:
|
| 166 |
+
in_channels: 1
|
| 167 |
+
out_channels: 1
|
| 168 |
+
layers_per_block: 2
|
| 169 |
+
layers_mid_block: 2
|
| 170 |
+
patch_size: 256
|
| 171 |
use_fp16: true
|