From 334d07bf96e0dc1f2e21864b6a21816e2f26f003 Mon Sep 17 00:00:00 2001
From: mio <74481573+mio2333@users.noreply.github.com>
Date: Thu, 8 Jun 2023 23:39:06 +0800
Subject: [PATCH] Update make_captions.py

Append sys path for make_captions.py to load blip module in the same folder to fix the error when you don't run this script under the folder
---
 finetune/make_captions.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/finetune/make_captions.py b/finetune/make_captions.py
index 9e51037f3..b20c41068 100644
--- a/finetune/make_captions.py
+++ b/finetune/make_captions.py
@@ -3,6 +3,7 @@
 import os
 import json
 import random
+import sys
 
 from pathlib import Path
 from PIL import Image
@@ -11,6 +12,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
 import library.train_util as train_util