Merge pull request bmaltais#581 from mio2333/patch-1

Update make_captions.py
Vitalya-79 · Jun 12, 2023 · 89c3033 · 89c3033
2 parents 6417f5d + 334d07b
commit 89c3033
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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