Duplicated from Shitao/OmniGen
730f5fd
1
2
from .data import DatasetFromJson, TrainDataCollator from .loss import training_losses