From 5dcd3b454dc20ec87dae27161ac29c74a5659e90 Mon Sep 17 00:00:00 2001 From: Shuang Wu Date: Sat, 14 Jun 2025 13:09:15 +0000 Subject: [PATCH] Update direct3d-s2-v-1-0/config.yaml --- direct3d-s2-v-1-0/config.yaml | 10 ++++++++++ 1 file changed, 10 insertions(+) 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