ls_mlkit.flow_matching.base_fm module

class ls_mlkit.flow_matching.base_fm.BaseFlow(config: BaseFlowConfig, time_scheduler: FlowMatchingTimeScheduler)

Bases: BaseGenerativeModel

abstract method: prior_sampling, compute_loss, step, sampling, inpainting

class ls_mlkit.flow_matching.base_fm.BaseFlowConfig(ndim_micro_shape: int, n_discretization_steps: int, n_inference_steps: int = None, *args: list[Any], **kwargs: dict[Any, Any])

Bases: BaseGenerativeModelConfig