Upload adv_d2_phtrandom_uniform_a0.0_ew0.10/configs/embedder.yaml with huggingface_hub

adv_d2_phtrandom_uniform_a0.0_ew0.10/configs/embedder.yaml

@@ -0,0 +1,111 @@
+model: unet_small2_yuv_quant
+
+hidden:
+  num_blocks: 4
+  num_bits: 16
+  in_channels: 3
+  out_channels: 3
+  z_channels: 64
+  normalization: 'group'
+  activation: 'gelu'
+  last_tanh: True
+
+hidden_orig:
+  num_blocks: 4
+  num_bits: 16
+  in_channels: 3
+  out_channels: 3
+  z_channels: 64
+  normalization: 'batch'
+  activation: 'relu'
+  last_tanh: False
+
+unet_small2_quant: # quantization friendly unet2
+  msg_processor:
+    nbits: 16
+    hidden_size: 32     # nbits * 2
+    msg_processor_type: 'binary+concat'
+  unet:
+    in_channels: 3
+    out_channels: 3
+    z_channels: 16
+    num_blocks: 8
+    activation: 'relu'
+    normalization: 'batch'
+    z_channels_mults: [1, 2, 4, 8]
+    last_tanh: True
+
+unet_small2_yuv_quant:
+  msg_processor:
+    nbits: 16
+    hidden_size: 32     # nbits * 2
+    msg_processor_type: 'binary+concat'
+  unet:
+    in_channels: 1
+    out_channels: 1
+    z_channels: 16
+    num_blocks: 8
+    activation: 'relu'
+    normalization: 'batch'
+    z_channels_mults: [1, 2, 4, 8]
+    last_tanh: True
+
+unet_base_yuv_quant:
+  msg_processor:
+    nbits: 16
+    hidden_size: 32     # nbits * 2
+    msg_processor_type: 'binary+concat'
+  unet:
+    in_channels: 1
+    out_channels: 1
+    z_channels: 16
+    num_blocks: 8
+    activation: 'relu'
+    normalization: 'batch'
+    z_channels_mults: [2, 4, 8, 16]
+    last_tanh: True
+
+
+unet_tiny_yuv_quant:
+  msg_processor:
+    nbits: 16
+    hidden_size: 32     # nbits * 2
+    msg_processor_type: 'binary+concat'
+  unet:
+    in_channels: 1
+    out_channels: 1
+    z_channels: 8
+    num_blocks: 4
+    activation: 'relu'
+    normalization: 'batch'
+    z_channels_mults: [1, 1, 2, 4, 8]
+    last_tanh: True
+
+#chunkyseal embedder
+unet_chunky: # VideoSeal with more capacity after message
+  msg_processor:
+    msg_processor_type: 'binary+concat'
+  unet:
+    in_channels: 3
+    out_channels: 3
+    z_channels: 16
+    num_blocks: 8
+    activation: 'relu'
+    normalization: 'batch'
+    z_channels_mults: [4, 8, 16, 32]
+    last_tanh: True
+
+# Alias for backward compatibility with old checkpoints
+unet_sweep_3: # Same as unet_chunky
+  msg_processor:
+    msg_processor_type: 'binary+concat'
+  unet:
+    in_channels: 3
+    out_channels: 3
+    z_channels: 16
+    num_blocks: 8
+    activation: 'relu'
+    normalization: 'batch'
+    z_channels_mults: [4, 8, 16, 32]
+    last_tanh: True
+