From 18b01a5aed42f5059345fc314ccf135bf2fe590b Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Wed, 31 Jan 2024 12:07:59 -0800 Subject: [PATCH] better path --- modules_forge/initialization.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/modules_forge/initialization.py b/modules_forge/initialization.py index 0cd86531..b5619f8a 100644 --- a/modules_forge/initialization.py +++ b/modules_forge/initialization.py @@ -1,3 +1,4 @@ +import os import sys @@ -24,6 +25,16 @@ def initialize_forge(): torch.zeros((1, 1)).to(device, torch.float32) model_management.soft_empty_cache() + import ldm_patched.utils.path_utils as ldm_patched_path_utils + from modules.paths import models_path, data_path + + ldm_patched_path_utils.base_path = data_path + ldm_patched_path_utils.models_dir = models_path + ldm_patched_path_utils.output_directory = os.path.join(data_path, "output") + ldm_patched_path_utils.temp_directory = os.path.join(data_path, "temp") + ldm_patched_path_utils.input_directory = os.path.join(data_path, "input") + ldm_patched_path_utils.user_directory = os.path.join(data_path, "user") + import modules_forge.patch_basic modules_forge.patch_basic.patch_all_basics()