From 0f42ab78c430166b773c4a5e0fb8c6d50fc68547 Mon Sep 17 00:00:00 2001
From: bmaltais <bernard@ducourier.com>
Date: Wed, 28 Dec 2022 23:19:18 -0500
Subject: [PATCH] Fix issue with caption extension

---
 lora_gui.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/lora_gui.py b/lora_gui.py
index fadc3ea..0eaa8cc 100644
--- a/lora_gui.py
+++ b/lora_gui.py
@@ -236,7 +236,7 @@ def train_model(
     seed,
     num_cpu_threads_per_process,
     cache_latent,
-    caption_extention,
+    caption_extension,
     enable_bucket,
     gradient_checkpointing,
     full_fp16,
@@ -396,8 +396,8 @@ def train_model(
     run_cmd += f' --seed={seed}'
     run_cmd += f' --save_precision={save_precision}'
     run_cmd += f' --logging_dir={logging_dir}'
-    if not caption_extention == '':
-        run_cmd += f' --caption_extention={caption_extention}'
+    if not caption_extension == '':
+        run_cmd += f' --caption_extension={caption_extension}'
     if not stop_text_encoder_training == 0:
         run_cmd += (
             f' --stop_text_encoder_training={stop_text_encoder_training}'