lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-26 09:18:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

Textual Inversion: Added custom training image size and number of repeats per input image in a single epoch

Browse Source
This commit is contained in:
alg-wiki
2022-10-10 17:07:46 +09:00
committed by AUTOMATIC1111
parent 8f1efdc130
commit ea00c1624b
4 changed files with 24 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modules/textual_inversion/preprocess.py
View File

@@ -7,8 +7,8 @@ import tqdm
from modules import shared, images
def preprocess(process_src, process_dst, process_flip, process_split, process_caption):
size = 512
def preprocess(process_src, process_dst, process_size, process_flip, process_split, process_caption):
size = process_size
src = os.path.abspath(process_src)
dst = os.path.abspath(process_dst)