Fix missing import in BLIP captioning

bmaltais · Dec 30, 2023 · e084bc5 · e084bc5
1 parent 89cfc46
commit e084bc5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/finetune/make_captions.py b/finetune/make_captions.py
@@ -13,7 +13,7 @@
 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")