neural-os

Runtime error

da03 commited on Nov 18, 2024

Commit

b871f44

1 Parent(s): b07d75e

.

Files changed (1) hide show

pssearch_bsz64_acc1_lr8e5_512_leftclick.yaml CHANGED Viewed

@@ -22,7 +22,7 @@ model:
       target: ldm.modules.diffusionmodules.openaimodel.UNetModel
       params:
         image_size: 64
-        in_channels: 34
         out_channels: 4
         model_channels: 192
         attention_resolutions:

       target: ldm.modules.diffusionmodules.openaimodel.UNetModel
       params:
         image_size: 64
+        in_channels: 41
         out_channels: 4
         model_channels: 192
         attention_resolutions: