developy commited on
Commit
7dc9882
·
verified ·
1 Parent(s): e523dcd

Update apdepth/marigold_pipeline.py

Browse files
Files changed (1) hide show
  1. apdepth/marigold_pipeline.py +1 -2
apdepth/marigold_pipeline.py CHANGED
@@ -31,9 +31,8 @@ from diffusers import (
31
  DDIMScheduler,
32
  DiffusionPipeline,
33
  LCMScheduler,
34
- # UNet2DConditionModel,
35
  )
36
- from marigold.modules.unet_2d_condition import UNet2DConditionModel
37
  from diffusers.utils import BaseOutput
38
  from PIL import Image
39
  from torch.utils.data import DataLoader, TensorDataset
 
31
  DDIMScheduler,
32
  DiffusionPipeline,
33
  LCMScheduler,
34
+ UNet2DConditionModel,
35
  )
 
36
  from diffusers.utils import BaseOutput
37
  from PIL import Image
38
  from torch.utils.data import DataLoader, TensorDataset