diff --git a/ldm/invoke/txt2mask.py b/ldm/invoke/txt2mask.py index 01d93546e3..f9a731d1f6 100644 --- a/ldm/invoke/txt2mask.py +++ b/ldm/invoke/txt2mask.py @@ -29,7 +29,7 @@ work fine. import torch import numpy as np -from models.clipseg import CLIPDensePredT +from clipseg import CLIPDensePredT from einops import rearrange, repeat from PIL import Image from torchvision import transforms