Update pipeline.py

pipeline.py

@@ -21,7 +21,7 @@ class KarrasEDMConditionalPipeline(DiffusionPipeline):
 
     model_cpu_offload_seq = "unet"
 
-    def __init__(self, unet: UNet2DModel, scheduler: FixedEDMEulerScheduler) -> None:
+    def __init__(self, unet, scheduler) -> None:
         super().__init__()
 
         self.register_modules(