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/BrushNet_CA.py
+++ b/iopaint/model/power_paint/v2/BrushNet_CA.py
@@ -3,7 +3,7 @@ from typing import Any, Dict, List, Optional, Tuple, Union

 import torch
 from diffusers import UNet2DConditionModel
-from diffusers.models.unet_2d_blocks import (
+from diffusers.models.unets.unet_2d_blocks import (
    get_down_block,
    get_mid_block,
    get_up_block,