diff --git a/modules/cmd_args.py b/modules/cmd_args.py index 04569ff4..07044184 100644 --- a/modules/cmd_args.py +++ b/modules/cmd_args.py @@ -25,6 +25,7 @@ parser.add_argument("--config", type=normalized_filepath, default=sd_default_con parser.add_argument("--ckpt", type=normalized_filepath, default=sd_model_file, help="path to checkpoint of stable diffusion model; if specified, this checkpoint will be added to the list of checkpoints and loaded",) parser.add_argument("--ckpt-dir", type=normalized_filepath, default=None, help="Path to directory with stable diffusion checkpoints") parser.add_argument("--vae-dir", type=normalized_filepath, default=None, help="Path to directory with VAE files") +parser.add_argument("--text-encoder-dir", type=normalized_filepath, default=None, help="Path to directory with text encoder models") parser.add_argument("--gfpgan-dir", type=normalized_filepath, help="GFPGAN directory", default=('./src/gfpgan' if os.path.exists('./src/gfpgan') else './GFPGAN')) parser.add_argument("--gfpgan-model", type=normalized_filepath, help="GFPGAN model file name", default=None) parser.add_argument("--no-half", action='store_true', help="do not switch the model to 16-bit floats") @@ -55,7 +56,7 @@ parser.add_argument("--esrgan-models-path", type=normalized_filepath, help="Path parser.add_argument("--bsrgan-models-path", type=normalized_filepath, help="Path to directory with BSRGAN model file(s).", default=os.path.join(models_path, 'BSRGAN')) parser.add_argument("--realesrgan-models-path", type=normalized_filepath, help="Path to directory with RealESRGAN model file(s).", default=os.path.join(models_path, 'RealESRGAN')) parser.add_argument("--dat-models-path", type=normalized_filepath, help="Path to directory with DAT model file(s).", default=os.path.join(models_path, 'DAT')) -parser.add_argument("--clip-models-path", type=normalized_filepath, help="Path to directory with CLIP model file(s).", default=None) +parser.add_argument("--clip-models-path", type=normalized_filepath, help="Path to directory with CLIP model file(s), for Interrogate options.", default=None) parser.add_argument("--xformers", action='store_true', help="enable xformers for cross attention layers") parser.add_argument("--force-enable-xformers", action='store_true', help="enable xformers for cross attention layers regardless of whether the checking code thinks you can run it; do not make bug reports if this fails to work") parser.add_argument("--xformers-flash-attention", action='store_true', help="enable xformers with Flash Attention to improve reproducibility (supported for SD2.x or variant only)") diff --git a/modules_forge/main_entry.py b/modules_forge/main_entry.py index 71164bb3..7ac7b853 100644 --- a/modules_forge/main_entry.py +++ b/modules_forge/main_entry.py @@ -150,6 +150,8 @@ def refresh_models(): if isinstance(shared.cmd_opts.vae_dir, str): module_paths.append(os.path.abspath(shared.cmd_opts.vae_dir)) + if isinstance(shared.cmd_opts.text_encoder_dir, str): + module_paths.append(os.path.abspath(shared.cmd_opts.text_encoder_dir)) for vae_path in module_paths: vae_files = find_files_with_extensions(vae_path, file_extensions)