Merge pull request #1821 from paboum/master

Fix missing import in BLIP captioning
bmaltais · Jan 3, 2024 · a2eed05 · a2eed05
2 parents a2c18df + e084bc5
commit a2eed05
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")