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

Fix to show template type

Browse Source
This commit is contained in:
Kohya S
2023-07-10 22:28:26 +09:00
parent f54b784d88
commit 68ca0ea995
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
train_textual_inversion.py
View File

@@ -94,7 +94,7 @@ class TextualInversionTrainer:
def assert_token_string(self, token_string, tokenizers): def assert_token_string(self, token_string, tokenizers):
pass pass
def get_text_cond(self, args, accelerator, batch, tokenizers, text_encoders, weight_dtype): def get_text_cond(self, args, accelerator, batch, tokenizers, text_encoders, weight_dtype):
with torch.enable_grad(): with torch.enable_grad():
input_ids = batch["input_ids"].to(accelerator.device) input_ids = batch["input_ids"].to(accelerator.device)
@@ -311,7 +311,7 @@ class TextualInversionTrainer:
# make captions: tokenstring tokenstring1 tokenstring2 ...tokenstringn という文字列に書き換える超乱暴な実装 # make captions: tokenstring tokenstring1 tokenstring2 ...tokenstringn という文字列に書き換える超乱暴な実装
if use_template: if use_template:
accelerator.print("use template for training captions. is object: {args.use_object_template}") accelerator.print(f"use template for training captions. is object: {args.use_object_template}")
templates = imagenet_templates_small if args.use_object_template else imagenet_style_templates_small templates = imagenet_templates_small if args.use_object_template else imagenet_style_templates_small
replace_to = " ".join(token_strings) replace_to = " ".join(token_strings)
captions = [] captions = []

2
train_textual_inversion_XTI.py
View File

@@ -234,7 +234,7 @@ def train(args):
# make captions: tokenstring tokenstring1 tokenstring2 ...tokenstringn という文字列に書き換える超乱暴な実装 # make captions: tokenstring tokenstring1 tokenstring2 ...tokenstringn という文字列に書き換える超乱暴な実装
if use_template: if use_template:
print("use template for training captions. is object: {args.use_object_template}") print(f"use template for training captions. is object: {args.use_object_template}")
templates = imagenet_templates_small if args.use_object_template else imagenet_style_templates_small templates = imagenet_templates_small if args.use_object_template else imagenet_style_templates_small
replace_to = " ".join(token_strings) replace_to = " ".join(token_strings)
captions = [] captions = []