Diffusion gemma: fix failed test cases (#47025)