请问支持C2I吗

[black-yt]

非常棒的工作，请问支持C2I模型的训练吗，比如DiT和SiT

[Jayce-Ping]

目前的repo不直接支持，但是应该可以很容易纳入框架从而复用已有的功能。关键在于实现sampling和forward的对应方法，更多可以查看guidance/new_model，里面提到了具体的步骤和示例。