From d92c4d5ab703746cba2e7fff4fa4441066eee9c8 Mon Sep 17 00:00:00 2001 From: Xie Zejian Date: Thu, 27 Apr 2023 22:39:14 +0800 Subject: [PATCH] fix typo in score sde pipeline (#3132) --- src/diffusers/pipelines/score_sde_ve/pipeline_score_sde_ve.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/pipelines/score_sde_ve/pipeline_score_sde_ve.py b/src/diffusers/pipelines/score_sde_ve/pipeline_score_sde_ve.py index 60a6f1e70f..3ff7b8ee46 100644 --- a/src/diffusers/pipelines/score_sde_ve/pipeline_score_sde_ve.py +++ b/src/diffusers/pipelines/score_sde_ve/pipeline_score_sde_ve.py @@ -33,7 +33,7 @@ class ScoreSdeVePipeline(DiffusionPipeline): unet: UNet2DModel scheduler: ScoreSdeVeScheduler - def __init__(self, unet: UNet2DModel, scheduler: DiffusionPipeline): + def __init__(self, unet: UNet2DModel, scheduler: ScoreSdeVeScheduler): super().__init__() self.register_modules(unet=unet, scheduler=scheduler)