@@ -33,6 +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")
gr.HTML(value="<p style='margin-bottom: 0.7em'>Train an embedding or Hypernetwork; you must specify a directory with a set of 1:1 ratio images <a href=\"https://github.com/AUTOMATIC1111/stable-diffusion-webui/wiki/Textual-Inversion\" style=\"font-weight:bold;\">[wiki]</a></p>")
withFormRow():
...
...
@@ -1345,7 +1348,11 @@ def create_ui():
dataset_directory=gr.Textbox(label='Dataset directory',placeholder="Path to directory with input images",elem_id="train_dataset_directory")
log_directory=gr.Textbox(label='Log directory',placeholder="Path to directory where to write outputs",value="textual_inversion",elem_id="train_log_directory")