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