
get_order_list

<
source
>
(
num_inference_steps: int

)


Parameters 

num_inference_steps (int) —
the number of diffusion steps used when generating samples with a pre-trained model.



Computes the solver order at each time step.

scale_model_input

<
source
>
(
sample: FloatTensor
*args
**kwargs

)
→
torch.FloatTensor

Parameters 

sample (torch.FloatTensor) — input sample


Returns

torch.FloatTensor



scaled input sample


Ensures inter