Issue with Generating Watermarked Image from Trained Model

I have successfully trained the model and obtained the checkpoint.pt file. However, upon attempting to generate an image using this checkpoint, the resulting image does not contain the expected watermark.

Could anyone kindly provide guidance on how to correctly generate an image with the watermark embedded? Any assistance or instructions on this matter would be greatly appreciated.