Duplicated from sudemai/OmniGen
55d914b
1
2
from .data import DatasetFromJson, TrainDataCollator from .loss import training_losses