Update apdepth/marigold_pipeline.py

apdepth/marigold_pipeline.py

@@ -116,7 +116,6 @@ class MarigoldPipeline(DiffusionPipeline):
         self,
         unet: UNet2DConditionModel,
         vae: AutoencoderKL,
-        scheduler: Union[DDIMScheduler, LCMScheduler],
         text_encoder: CLIPTextModel,
         tokenizer: CLIPTokenizer,
         scale_invariant: Optional[bool] = True,
@@ -128,7 +127,6 @@ class MarigoldPipeline(DiffusionPipeline):
         self.register_modules(
             unet=unet,
             vae=vae,
-            scheduler=scheduler,
             text_encoder=text_encoder,
             tokenizer=tokenizer,
         )
@@ -212,7 +210,7 @@ class MarigoldPipeline(DiffusionPipeline):
         assert processing_res >= 0
 
         # Check if denoising step is reasonable
-        self._check_inference_step(denoising_steps)
+        # self._check_inference_step(denoising_steps)
 
         resample_method: InterpolationMode = get_tv_resample_method(resample_method)