diff --git a/finetune/make_captions.py b/finetune/make_captions.py index f1b83b1..0716690 100644 --- a/finetune/make_captions.py +++ b/finetune/make_captions.py @@ -13,7 +13,7 @@ import torch from torchvision import transforms from torchvision.transforms.functional import InterpolationMode sys.path.append(os.path.join(os.path.dirname(__file__), '..')) # sys.path.append(os.path.dirname(__file__)) -from blip.blip import blip_decoder +from blip.blip import blip_decoder, is_url import library.train_util as train_util DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")