Update sde_ve_pipeline.py

sde_ve_pipeline.py

@@ -1,3 +1,5 @@
+from .unet.unet_model import UNet2DModel
+from .scheduler.sde_ve_scheduler import ScoreSdeVeScheduler
 from diffusers.utils.torch_utils import randn_tensor
 from diffusers.pipelines.pipeline_utils import DiffusionPipeline, ImagePipelineOutput
 
@@ -14,8 +16,8 @@ class ScoreSdeVePipeline(DiffusionPipeline):
             A `ScoreSdeVeScheduler` to be used in combination with `unet` to denoise the encoded image.
     """
 
-    unet
-    scheduler
+    unet: UNet2DModel
+    scheduler: ScoreSdeVeScheduler
 
     def __init__(self, unet, scheduler):
         super().__init__()