VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-09 06:45:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix path error

Browse Source
This commit is contained in:
CjangCjengh
2023-12-06 09:48:42 +08:00
parent 46cf41cc93
commit d653e594c2
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
finetune/make_captions.py
View File

@@ -13,7 +13,7 @@ import torch
from torchvision import transforms from torchvision import transforms
from torchvision.transforms.functional import InterpolationMode from torchvision.transforms.functional import InterpolationMode
sys.path.append(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 import library.train_util as train_util
DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu") DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
@@ -76,6 +76,8 @@ def main(args):
cwd = os.getcwd() cwd = os.getcwd()
print("Current Working Directory is: ", cwd) print("Current Working Directory is: ", cwd)
os.chdir("finetune") 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}") print(f"load images from {args.train_data_dir}")
train_data_dir_path = Path(args.train_data_dir) train_data_dir_path = Path(args.train_data_dir)