update LatentNoiseTrainer class to include iteration callbacks c6e7b6c verified fffiloni commited on Oct 11