update LatentNoiseTrainer class to include iteration callbacks c6e7b6c verified fffiloni commited on 5 days ago