From d653e594c278efb6c20d11b815778fcc50966f44 Mon Sep 17 00:00:00 2001
From: CjangCjengh <101577701+CjangCjengh@users.noreply.github.com>
Date: Wed, 6 Dec 2023 09:48:42 +0800
Subject: [PATCH] fix path error

---
 finetune/make_captions.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/finetune/make_captions.py b/finetune/make_captions.py
index b20c4106..61c766e0 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.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")
@@ -76,6 +76,8 @@ def main(args):
         cwd = os.getcwd()
         print("Current Working Directory is: ", cwd)
         os.chdir("finetune")
+        if not is_url(args.caption_weights):
+            args.caption_weights = os.path.join("..", args.caption_weights)
 
     print(f"load images from {args.train_data_dir}")
     train_data_dir_path = Path(args.train_data_dir)