fix deprecate warning

2024-11-23 18:58:34 +08:00
parent 4cec8ded64
commit 36e65f0097
3 changed files with 113 additions and 54 deletions
--- a/iopaint/model/power_paint/v2/unet_2d_condition.py
+++ b/iopaint/model/power_paint/v2/unet_2d_condition.py
@@ -15,7 +15,7 @@ from typing import Any, Dict, Optional, Tuple, Union

 import torch
 import torch.utils.checkpoint
-from diffusers.models.unet_2d_condition import UNet2DConditionOutput
+from diffusers.models.unets.unet_2d_condition import UNet2DConditionOutput
 from diffusers.utils import (
    USE_PEFT_BACKEND,
    deprecate,