lemonaddie commited on
Commit
d2a51cd
1 Parent(s): 9822b8a

Update models/geowizard_pipeline.py

Browse files
Files changed (1) hide show
  1. models/geowizard_pipeline.py +2 -0
models/geowizard_pipeline.py CHANGED
@@ -198,6 +198,8 @@ class DepthNormalEstimationPipeline(DiffusionPipeline):
198
 
199
  normal_colored = ((normal_pred + 1)/2 * 255).astype(np.uint8)
200
  normal_colored_img = Image.fromarray(normal_colored)
 
 
201
 
202
  return DepthNormalPipelineOutput(
203
  depth_np = depth_pred,
 
198
 
199
  normal_colored = ((normal_pred + 1)/2 * 255).astype(np.uint8)
200
  normal_colored_img = Image.fromarray(normal_colored)
201
+
202
+ self.img_embed = None
203
 
204
  return DepthNormalPipelineOutput(
205
  depth_np = depth_pred,