Spaces:
Build error
Build error
lemonaddie
commited on
Commit
•
db47fe5
1
Parent(s):
c818db8
Update models/depth_normal_pipeline_clip.py
Browse files
models/depth_normal_pipeline_clip.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 |
processing_res: int = 768,
|
79 |
match_input_res:bool =True,
|
@@ -95,7 +95,7 @@ class DepthNormalEstimationPipeline(DiffusionPipeline):
|
|
95 |
)," Value Error: `resize_output_back` is only valid with "
|
96 |
|
97 |
assert processing_res >=0
|
98 |
-
assert
|
99 |
assert ensemble_size >=1
|
100 |
|
101 |
# --------------- Image Processing ------------------------
|
@@ -145,7 +145,7 @@ class DepthNormalEstimationPipeline(DiffusionPipeline):
|
|
145 |
|
146 |
depth_pred_raw, normal_pred_raw = self.single_infer(
|
147 |
input_rgb=batched_image,
|
148 |
-
num_inference_steps=
|
149 |
domain=domain,
|
150 |
show_pbar=show_progress_bar,
|
151 |
)
|
|
|
73 |
@torch.no_grad()
|
74 |
def __call__(self,
|
75 |
input_image:Image,
|
76 |
+
denoising_steps: int = 10,
|
77 |
ensemble_size: int = 10,
|
78 |
processing_res: int = 768,
|
79 |
match_input_res:bool =True,
|
|
|
95 |
)," Value Error: `resize_output_back` is only valid with "
|
96 |
|
97 |
assert processing_res >=0
|
98 |
+
assert denoising_steps >=1
|
99 |
assert ensemble_size >=1
|
100 |
|
101 |
# --------------- Image Processing ------------------------
|
|
|
145 |
|
146 |
depth_pred_raw, normal_pred_raw = self.single_infer(
|
147 |
input_rgb=batched_image,
|
148 |
+
num_inference_steps=denoising_steps,
|
149 |
domain=domain,
|
150 |
show_pbar=show_progress_bar,
|
151 |
)
|