fix incorrect import of clipseg

This commit is contained in:
Lincoln Stein 2022-10-18 19:28:30 -04:00 committed by GitHub
parent 66f6ef1b35
commit 9b3ac92c24
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -29,7 +29,7 @@ work fine.
import torch import torch
import numpy as np import numpy as np
from models.clipseg import CLIPDensePredT from clipseg import CLIPDensePredT
from einops import rearrange, repeat from einops import rearrange, repeat
from PIL import Image from PIL import Image
from torchvision import transforms from torchvision import transforms