training Consistency Models

[Hedvigskorge]

Hello everyone,

Has anyone successfully utilized the CMStochasticIterativeScheduler for training a Consistency Model from scratch?

I've been exploring the pipeline_consistency_models.py, which relies on a pretrained model. However, I'm looking for examples or guidance on how to integrate this scheduler into a custom script for Consistency Training using my own dataset. particularly how to compute the loss function when integrating this scheduler.

Any insights or examples would be greatly appreciated!

[tolgacangoz]

Hello @Hedvigskorge,
Training examples can be found in the examples folder. Could you examine consistency-related files?