lemonaddie
commited on
Commit
•
f3be028
1
Parent(s):
9ce4745
Update models/depth_normal_pipeline_clip_cfg.py
Browse files
models/depth_normal_pipeline_clip_cfg.py
CHANGED
@@ -232,7 +232,7 @@ class DepthNormalEstimationPipeline(DiffusionPipeline):
|
|
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
|
|
|
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
|