#5

hako-mikan · Feb 7, 2024 · e24c79f · e24c79f
1 parent 4898480
commit e24c79f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/trainer/dataset.py b/trainer/dataset.py
@@ -285,6 +285,7 @@ def encode_image_text(t):
                     prompt = t.lora_trigger_word + ", " + caption
                 else:
                     prompt = t.lora_trigger_word
+                t.tagcount(prompt)
                 if "BASE" not in t.network_blocks:
                     emb1, emb2 = (emp1, emp2) if prompt is None else t.text_model.encode_text(prompt)
                 else: