diff --git a/modules/shared.py b/modules/shared.py
index ca2d9737b..da51bde00 100644
--- a/modules/shared.py
+++ b/modules/shared.py
@@ -33,7 +33,7 @@ parser.add_argument("--no-half-vae", action='store_true', help="do not switch th
 parser.add_argument("--no-progressbar-hiding", action='store_true', help="do not hide progressbar in gradio UI (we hide it because it slows down ML if you have hardware acceleration in browser)")
 parser.add_argument("--max-batch-count", type=int, default=16, help="maximum batch count value for the UI")
 parser.add_argument("--embeddings-dir", type=str, default=os.path.join(script_path, 'embeddings'), help="embeddings directory for textual inversion (default: embeddings)")
-parser.add_argument("--textual-inversion-templates-dir", type=str, default=os.path.join(script_path, 'textual_inversion_templates'), help="directory with textual inversion templates")
+parser.add_argument("--textual-inversion-templates-dir", type=str, default=os.path.join(script_path, 'train/templates'), help="directory with textual inversion templates")
 parser.add_argument("--hypernetwork-dir", type=str, default=os.path.join(models_path, 'hypernetworks'), help="hypernetwork directory")
 parser.add_argument("--localizations-dir", type=str, default=os.path.join(script_path, 'localizations'), help="localizations directory")
 parser.add_argument("--allow-code", action='store_true', help="allow custom script execution from webui")
diff --git a/ui-config.json b/ui-config.json
index 0eac8bd9d..6524484a0 100644
--- a/ui-config.json
+++ b/ui-config.json
@@ -666,5 +666,9 @@
     "customscript/additional_networks.py/img2img/Weight 5/value": 1,
     "customscript/additional_networks.py/img2img/Weight 5/minimum": -1.0,
     "customscript/additional_networks.py/img2img/Weight 5/maximum": 2.0,
-    "customscript/additional_networks.py/img2img/Weight 5/step": 0.05
+    "customscript/additional_networks.py/img2img/Weight 5/step": 0.05,
+    "train/Prompt template/visible": true,
+    "train/Prompt template/value": "subject_filewords.txt",
+    "train/Do not resize images/visible": true,
+    "train/Do not resize images/value": false
 }
\ No newline at end of file