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
- denosing_steps: int = 10,
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 denosing_steps >=1
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=denosing_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
  )