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
         |