seamlesswatermark.yaml 921 B

1234567891011121314151617181920212223242526272829303132333435363738394041424344
  1. # Copyright (c) Meta Platforms, Inc. and affiliates.
  2. # All rights reserved.
  3. #
  4. # This source code is licensed under the BSD-style license found in the
  5. # LICENSE file in the root directory of this source tree.
  6. name: seamlesswatermark
  7. model_type: seanet
  8. checkpoint: "/large_experiments/seamless/nllb/watermarking/checkpoints/ckpt_e9d0008c.th"
  9. watermarker_model:
  10. channels: 1
  11. sample_rate: 16000
  12. seanet:
  13. activation: ELU
  14. activation_params:
  15. alpha: 1.0
  16. causal: false
  17. channels: 1
  18. compress: 2
  19. decoder:
  20. final_activation: null
  21. final_activation_params: null
  22. trim_right_ratio: 1.0
  23. detector: {}
  24. dilation_base: 2
  25. dimension: 128
  26. disable_norm_outer_blocks: 0
  27. encoder: {}
  28. kernel_size: 7
  29. last_kernel_size: 7
  30. lstm: 2
  31. n_filters: 32
  32. n_residual_layers: 1
  33. norm: weight_norm
  34. norm_params: {}
  35. pad_mode: constant
  36. ratios:
  37. - 8
  38. - 5
  39. - 4
  40. - 2
  41. residual_kernel_size: 3
  42. true_skip: true