fix multiscale discr logits

[lucidrains]

thanks! haven't added all the multiscale discriminator code just yet, but should be done this week