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

Merge branch 'main' into original-u-net

Browse Source
This commit is contained in:
Kohya S
2023-06-14 12:49:51 +09:00
parent 67f09b7d7e 89c3033401
commit 44404fcd6d
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
finetune/make_captions.py
View File

@@ -3,6 +3,7 @@ import glob
import os import os
import json import json
import random import random
import sys
from pathlib import Path from pathlib import Path
from PIL import Image from PIL import Image
@@ -11,6 +12,7 @@ import numpy as np
import torch 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__))
from blip.blip import blip_decoder from blip.blip import blip_decoder
import library.train_util as train_util import library.train_util as train_util