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