Unverified Commit d92c4d5a authored by Xie Zejian's avatar Xie Zejian Committed by GitHub
Browse files

fix typo in score sde pipeline (#3132)

parent eade4308
......@@ -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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment