wushuang98 commited on
Commit
8b04a8e
·
verified ·
1 Parent(s): 5dcd3b4

Update direct3d-s2-v-1-1/config.yaml

Browse files
Files changed (1) hide show
  1. direct3d-s2-v-1-1/config.yaml +10 -0
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