Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
9d1914d
1
Parent(s):
95c09ba
Update models/depth_normal_pipeline_clip_cfg.py
Browse files
models/depth_normal_pipeline_clip_cfg.py
CHANGED
@@ -148,6 +148,7 @@ class DepthNormalEstimationPipeline(DiffusionPipeline):
|
|
148 |
input_rgb=batched_image,
|
149 |
num_inference_steps=denoising_steps,
|
150 |
domain=domain,
|
|
|
151 |
show_pbar=show_progress_bar,
|
152 |
)
|
153 |
depth_pred_ls.append(depth_pred_raw.detach().clone())
|
@@ -231,6 +232,7 @@ class DepthNormalEstimationPipeline(DiffusionPipeline):
|
|
231 |
def single_infer(self,input_rgb:torch.Tensor,
|
232 |
num_inference_steps:int,
|
233 |
domain:str,
|
|
|
234 |
show_pbar:bool,):
|
235 |
|
236 |
device = input_rgb.device
|
|
|
148 |
input_rgb=batched_image,
|
149 |
num_inference_steps=denoising_steps,
|
150 |
domain=domain,
|
151 |
+
guidance_scale=guidance_scale
|
152 |
show_pbar=show_progress_bar,
|
153 |
)
|
154 |
depth_pred_ls.append(depth_pred_raw.detach().clone())
|
|
|
232 |
def single_infer(self,input_rgb:torch.Tensor,
|
233 |
num_inference_steps:int,
|
234 |
domain:str,
|
235 |
+
guidance_scale: int
|
236 |
show_pbar:bool,):
|
237 |
|
238 |
device = input_rgb.device
|