lemonaddie commited on
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