developy commited on
Commit
41fff5f
·
verified ·
1 Parent(s): 9f63d59

Update apdepth/marigold_pipeline.py

Browse files
Files changed (1) hide show
  1. apdepth/marigold_pipeline.py +1 -1
apdepth/marigold_pipeline.py CHANGED
@@ -394,7 +394,7 @@ class MarigoldPipeline(DiffusionPipeline):
394
  """
395
  device = self.device
396
  rgb_in = rgb_in.to(device)
397
- depth_da2 = self.da2.infer_batch(rgb_in).to(device)
398
 
399
  with torch.no_grad():
400
  # Encode image
 
394
  """
395
  device = self.device
396
  rgb_in = rgb_in.to(device)
397
+ depth_da2 = self.da2.infer_batch(rgb_in).to(device)
398
 
399
  with torch.no_grad():
400
  # Encode image