Update apdepth/marigold_pipeline.py
Browse files
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 =
|
| 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",
|