_partial_: true
_target_: models.losses.RiemannianFlowMatchingLoss
cond_drop_rate: 0.0
conditioning_key: ${model.cond_preprocessing.output_key}