developy commited on
Commit
587b709
·
verified ·
1 Parent(s): 4c8da60

Update apdepth/marigold_pipeline.py

Browse files
Files changed (1) hide show
  1. apdepth/marigold_pipeline.py +1 -1
apdepth/marigold_pipeline.py CHANGED
@@ -165,7 +165,7 @@ class MarigoldPipeline(DiffusionPipeline):
165
  self,
166
  input_image: Union[Image.Image, torch.Tensor],
167
  denoising_steps: Optional[int] = None,
168
- ensemble_size: int = 5,
169
  processing_res: Optional[int] = None,
170
  match_input_res: bool = True,
171
  resample_method: str = "bilinear",
 
165
  self,
166
  input_image: Union[Image.Image, torch.Tensor],
167
  denoising_steps: Optional[int] = None,
168
+ ensemble_size: int = 1,
169
  processing_res: Optional[int] = None,
170
  match_input_res: bool = True,
171
  resample_method: str = "bilinear",