minor fix to add_noise method for mix precision policy

[costiash]

Encountering into exception when running with mixed precision policy and passing 'init_image' argument into the generator, so added casting in add_noise method what made it work without any errors.

[matpalm]

+1. just did the same thing, in a different way!