Skip to content

Commit d92c4d5

Browse files
authored
fix typo in score sde pipeline (#3132)
1 parent eade430 commit d92c4d5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/pipelines/score_sde_ve/pipeline_score_sde_ve.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ class ScoreSdeVePipeline(DiffusionPipeline):
3333
unet: UNet2DModel
3434
scheduler: ScoreSdeVeScheduler
3535

36-
def __init__(self, unet: UNet2DModel, scheduler: DiffusionPipeline):
36+
def __init__(self, unet: UNet2DModel, scheduler: ScoreSdeVeScheduler):
3737
super().__init__()
3838
self.register_modules(unet=unet, scheduler=scheduler)
3939

0 commit comments

Comments
 (0)