diff --git a/direct3d-s2-v-1-0/config.yaml b/direct3d-s2-v-1-0/config.yaml index 2e3f094..6f36c3e 100644 --- a/direct3d-s2-v-1-0/config.yaml +++ b/direct3d-s2-v-1-0/config.yaml @@ -158,4 +158,14 @@ refiner: layers_per_block: 2 layers_mid_block: 2 patch_size: 192 + use_fp16: true + +refiner_1024: + target: direct3d_s2.models.refiner.unet_refiner.Voxel_RefinerXL_sign + params: + in_channels: 1 + out_channels: 1 + layers_per_block: 2 + layers_mid_block: 2 + patch_size: 256 use_fp16: true \ No newline at end of file