diff --git a/modules/cmd_args.py b/modules/cmd_args.py index fe37814e9..92b3c1bc8 100644 --- a/modules/cmd_args.py +++ b/modules/cmd_args.py @@ -13,7 +13,7 @@ group.add_argument("--medvram", action='store_true', help="Split model stages an group.add_argument("--lowvram", action='store_true', help="Split model components and keep only active part in VRAM, default: %(default)s") group.add_argument("--ckpt", type=str, default=None, help="Path to model checkpoint to load immediately, default: %(default)s") group.add_argument('--vae', type=str, default=None, help='Path to VAE checkpoint to load immediately, default: %(default)s') -group.add_argument("--data-dir", type=str, default=os.path.dirname(os.path.dirname(os.path.realpath(__file__))), help="Base path where all user data is stored, default: %(default)s") +group.add_argument("--data-dir", type=str, default="", help="Base path where all user data is stored, default: %(default)s") group.add_argument("--models-dir", type=str, default="models", help="Base path where all models are stored, default: %(default)s",) group.add_argument("--allow-code", action='store_true', help="Allow custom script execution, default: %(default)s") group.add_argument("--share", action='store_true', help="Enable UI accessible through Gradio site, default: %(default)s") diff --git a/modules/modelloader.py b/modules/modelloader.py index 5430f6364..431ff3fb1 100644 --- a/modules/modelloader.py +++ b/modules/modelloader.py @@ -277,7 +277,6 @@ def load_models(model_path: str, model_url: str = None, command_path: str = None @return: A list of paths containing the desired model(s) """ places = unique_directories([model_path, command_path]) - #shared.log.debug(f"{inspect.currentframe().f_code.co_name}: {', '.join(places)}") output = [] try: output:list = [*filter(extension_filter(ext_filter, ext_blacklist), directory_files(*places))] diff --git a/modules/paths.py b/modules/paths.py index 7ff14ee3d..1943bca63 100644 --- a/modules/paths.py +++ b/modules/paths.py @@ -63,7 +63,7 @@ def create_paths(opts): def fix_path(folder): if opts.data.get(folder, None) is None or opts.data[folder] is None or opts.data[folder] == '': return - if os.path.isabs(opts.data[folder]) or opts.data[folder].startswith(data_path): + if os.path.isabs(opts.data[folder]) or (len(data_path) > 0 and opts.data[folder].startswith(data_path)) and not opts.data[folder].startswith(script_path): return fullpath = os.path.join(data_path, opts.data[folder]) relpath = os.path.relpath(fullpath, script_path) diff --git a/modules/paths_internal.py b/modules/paths_internal.py index 3c2ce7fba..44afc4ad3 100644 --- a/modules/paths_internal.py +++ b/modules/paths_internal.py @@ -12,7 +12,7 @@ default_sd_model_file = sd_model_file # Parse the --data-dir flag first so we can use it as a base for our other argument default values parser_pre = argparse.ArgumentParser(add_help=False) -parser_pre.add_argument("--data-dir", type=str, default=os.path.dirname(modules_path), help="base path where all user data is stored", ) +parser_pre.add_argument("--data-dir", type=str, default="", help="base path where all user data is stored", ) parser_pre.add_argument("--models-dir", type=str, default="models", help="base path where all models are stored",) cmd_opts_pre = parser_pre.parse_known_args()[0] data_path = cmd_opts_pre.data_dir diff --git a/modules/processing_diffusers.py b/modules/processing_diffusers.py index 54a709899..55b40f6d5 100644 --- a/modules/processing_diffusers.py +++ b/modules/processing_diffusers.py @@ -266,7 +266,6 @@ def process_diffusers(p: StableDiffusionProcessing, seeds, prompts, negative_pro if shared.opts.save and not p.do_not_save_samples and shared.opts.save_images_before_highres_fix and hasattr(shared.sd_model, 'vae'): save_intermediate(latents=output.images, suffix="-before-hires") hires_resize(latents=output.images) - print('HERE', p.init_images) sd_models.set_diffuser_pipe(shared.sd_model, sd_models.DiffusersTaskType.IMAGE_2_IMAGE) p.ops.append('hires') hires_args = set_pipeline_args( diff --git a/modules/ui_extra_networks.py b/modules/ui_extra_networks.py index 3ef1937b1..e9e8df7f9 100644 --- a/modules/ui_extra_networks.py +++ b/modules/ui_extra_networks.py @@ -113,7 +113,8 @@ class ExtraNetworksPage: def link_preview(self, filename): quoted_filename = urllib.parse.quote(filename.replace('\\', '/')) mtime = os.path.getmtime(filename) - return f"./sd_extra_networks/thumb?filename={quoted_filename}&mtime={mtime}" + preview = f"./sd_extra_networks/thumb?filename={quoted_filename}&mtime={mtime}" + return preview def search_terms_from_path(self, filename, possible_directories=None): abspath = os.path.abspath(filename) @@ -158,16 +159,16 @@ class ExtraNetworksPage: return f"