We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1bd6c0e commit 15040b0Copy full SHA for 15040b0
pipelines/score_sde_ve/pipeline_score_sde_ve.py
@@ -33,7 +33,7 @@ class ScoreSdeVePipeline(DiffusionPipeline):
33
unet: UNet2DModel
34
scheduler: ScoreSdeVeScheduler
35
36
- def __init__(self, unet: UNet2DModel, scheduler: DiffusionPipeline):
+ def __init__(self, unet: UNet2DModel, scheduler: ScoreSdeVeScheduler):
37
super().__init__()
38
self.register_modules(unet=unet, scheduler=scheduler)
39
0 commit comments