Update apdepth/marigold_pipeline.py
Browse files
apdepth/marigold_pipeline.py
CHANGED
|
@@ -31,9 +31,8 @@ from diffusers import (
|
|
| 31 |
DDIMScheduler,
|
| 32 |
DiffusionPipeline,
|
| 33 |
LCMScheduler,
|
| 34 |
-
|
| 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
|