Update conditional_pipeline.py

conditional_pipeline.py

@@ -18,7 +18,7 @@ class ScoreSdeVePipelineConditioned(DiffusionPipeline):
         scheduler ([`ScoreSdeVeScheduler`]):
             A `ScoreSdeVeScheduler` to be used in combination with `unet` to denoise the encoded image.
     """
-    def __init__(self, unet: UNet2DModel, scheduler: ScoreSdeVeScheduler):
+    def __init__(self, unet, scheduler):
         super().__init__()
         self.register_modules(unet=unet, scheduler=scheduler)