_partial_: true
_target_: models.samplers.riemannian_flow_sampler.riemannian_flow_sampler
num_steps: 250
cfg_rate: ${model.cfg_rate}