From 8b04a8eddb7a56a0f4e89fe5f5b840c7d5610c00 Mon Sep 17 00:00:00 2001
From: Shuang Wu <wushuang98@users.noreply.huggingface.co>
Date: Sat, 14 Jun 2025 13:09:30 +0000
Subject: [PATCH] Update direct3d-s2-v-1-1/config.yaml

---
 direct3d-s2-v-1-1/config.yaml | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/direct3d-s2-v-1-1/config.yaml b/direct3d-s2-v-1-1/config.yaml
index 930585c..be6cb15 100644
--- a/direct3d-s2-v-1-1/config.yaml
+++ b/direct3d-s2-v-1-1/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