flowmatch_set_timesteps

Set timesteps for inference

Description

Configures the scheduler timesteps for a specific number of inference steps. This must be called before using the scheduler for denoising.

Usage

flowmatch_set_timesteps(schedule, num_inference_steps = 50L, device = "cpu",
                        mu = NULL, sigmas = NULL, timesteps = NULL)

Arguments

  • schedule: List. The FlowMatch scheduler object.
  • num_inference_steps: Integer. Number of denoising steps. Default: 50
  • device: Character or torch device. Device for tensors. Default: “cpu”
  • mu: Numeric or NULL. Shift parameter for dynamic shifting. Required if use_dynamic_shifting is TRUE. Default: NULL
  • sigmas: Numeric vector or NULL. Custom sigma values. Default: NULL
  • timesteps: Numeric vector or NULL

Value

Updated scheduler with configured timesteps and sigmas.