import watermark_codec.model_encoder import watermark_codec.model_decoder ModelEncoder = watermark_codec.model_encoder.ModelEncoder ModelDecoder = watermark_codec.model_decoder.ModelDecoder