lemonaddie
commited on
Update models/depth_normal_pipeline_clip_cfg.py
Browse files
models/depth_normal_pipeline_clip_cfg.py
CHANGED
@@ -73,7 +73,7 @@ class DepthNormalEstimationPipeline(DiffusionPipeline):
|
|
73 |
@torch.no_grad()
|
74 |
def __call__(self,
|
75 |
input_image:Image,
|
76 |
-
|
77 |
ensemble_size: int = 10,
|
78 |
guidance_scale: int = 1,
|
79 |
processing_res: int = 768,
|
@@ -96,7 +96,7 @@ class DepthNormalEstimationPipeline(DiffusionPipeline):
|
|
96 |
)," Value Error: `resize_output_back` is only valid with "
|
97 |
|
98 |
assert processing_res >=0
|
99 |
-
assert
|
100 |
assert ensemble_size >=1
|
101 |
|
102 |
# --------------- Image Processing ------------------------
|
@@ -146,7 +146,7 @@ class DepthNormalEstimationPipeline(DiffusionPipeline):
|
|
146 |
|
147 |
depth_pred_raw, normal_pred_raw = self.single_infer(
|
148 |
input_rgb=batched_image,
|
149 |
-
num_inference_steps=
|
150 |
domain=domain,
|
151 |
show_pbar=show_progress_bar,
|
152 |
)
|
|
|
73 |
@torch.no_grad()
|
74 |
def __call__(self,
|
75 |
input_image:Image,
|
76 |
+
denoising_steps: int = 10,
|
77 |
ensemble_size: int = 10,
|
78 |
guidance_scale: int = 1,
|
79 |
processing_res: int = 768,
|
|
|
96 |
)," Value Error: `resize_output_back` is only valid with "
|
97 |
|
98 |
assert processing_res >=0
|
99 |
+
assert denoising_steps >=1
|
100 |
assert ensemble_size >=1
|
101 |
|
102 |
# --------------- Image Processing ------------------------
|
|
|
146 |
|
147 |
depth_pred_raw, normal_pred_raw = self.single_infer(
|
148 |
input_rgb=batched_image,
|
149 |
+
num_inference_steps=denoising_steps,
|
150 |
domain=domain,
|
151 |
show_pbar=show_progress_bar,
|
152 |
)
|