Diffusers documentation
ConsistencyDecoderScheduler
ConsistencyDecoderScheduler
此调度器是 ConsistencyDecoderPipeline 的一部分，并在 DALL-E 3 中引入。
原始代码库可在 openai/consistency_models 中找到。
ConsistencyDecoderScheduler
class diffusers.schedulers.ConsistencyDecoderScheduler
< source >( num_train_timesteps: int = 1024sigma_data: float = 0.5 )
用于 Stable Diffusion 流水线中一致性解码器的调度器。
该调度器使用一致性模型实现两步去噪过程，将潜在表示解码为图像。
该模型继承自 SchedulerMixin 和 ConfigMixin。有关该库为所有调度器实现的通用方法（例如加载和保存），请参阅父类文档。
scale_model_input
< source >( sample: Tensortimestep: int | None = None ) → torch.Tensor
确保与那些需要根据当前时间步对去噪模型输入进行缩放的调度器之间的可互换性。
步骤
< source >( model_output: Tensortimestep: typing.Union[float, torch.Tensor]sample: Tensorgenerator: typing.Optional[torch.Generator] = Nonereturn_dict: bool = True ) → ConsistencyDecoderSchedulerOutput or tuple
参数
- model_output (
torch.Tensor) — 学习到的扩散模型的直接输出。 - timestep (
float或torch.Tensor) — 扩散链中的当前时间步。 - sample (
torch.Tensor) — 扩散过程生成的样本的当前实例。 - generator (
torch.Generator, 可选) — 用于确保可复现性的随机数生成器。 - return_dict (
bool，可选，默认为True) — 是否返回ConsistencyDecoderSchedulerOutput或tuple。
返回值
ConsistencyDecoderSchedulerOutput 或 tuple
如果 return_dict 为 True，则返回 ConsistencyDecoderSchedulerOutput，否则返回一个元组，其中第一个元素是样本张量。
通过逆向 SDE 预测上一时间步的样本。此函数将扩散过程从学习到的模型输出（通常是预测的噪声）进行传播。