diff --git a/modules/build_prompt.py b/modules/build_prompt.py index bc5f869..2d839bd 100644 --- a/modules/build_prompt.py +++ b/modules/build_prompt.py @@ -971,7 +971,7 @@ class BuildPromptManager: def _init_build_prompt_manager(): script_dir = os.path.dirname(os.path.abspath(__file__)) - config_dir = os.path.join(script_dir, "../assets") + config_dir = os.path.join(script_dir, "..", "assets") return BuildPromptManager(config_dir, config) # 初始化 diff --git a/modules/config.py b/modules/config.py index e4d9252..d3ab68b 100644 --- a/modules/config.py +++ b/modules/config.py @@ -26,7 +26,7 @@ class Config: def _load_config(): script_dir = os.path.dirname(os.path.abspath(__file__)) - config_file = os.path.join(script_dir, "../assets", "config.json") + config_file = os.path.join(script_dir, "..", "assets", "config.json") return Config(config_file) diff --git a/nodes/prompt_generate.py b/nodes/prompt_generate.py index 9a9b93d..1edab67 100644 --- a/nodes/prompt_generate.py +++ b/nodes/prompt_generate.py @@ -53,7 +53,7 @@ class GeneratePrompt: classify_datas = buildClassifyDatas(classify) file_name = f"{classify_datas}.jsonl" - data_file = os.path.join(script_dir, "../assets", language_dir, file_name) + data_file = os.path.join(script_dir, "..", "assets", language_dir, file_name) build = BuildPrompt(data_file) prompt = build.generate_prompt(seed) diff --git a/nodes/prompt_negative.py b/nodes/prompt_negative.py index 5625c4e..251fff8 100644 --- a/nodes/prompt_negative.py +++ b/nodes/prompt_negative.py @@ -29,7 +29,7 @@ class NegativePrompt: # 获取Script的目录路径 script_dir = os.path.dirname(os.path.abspath(__file__)) # Script directory language_dir = "zh" if language == "Chinese" else "en" - data_file = os.path.join(script_dir, "../assets", language_dir, "negative_datas.jsonl") + data_file = os.path.join(script_dir, "..", "assets", language_dir, "negative_datas.jsonl") build = BuildPrompt(data_file) negative_prompt = build.generate_prompt(seed) diff --git a/nodes/prompt_scenery.py b/nodes/prompt_scenery.py index d6bab6a..e1115f0 100644 --- a/nodes/prompt_scenery.py +++ b/nodes/prompt_scenery.py @@ -69,7 +69,7 @@ class SceneryPrompt: # 获取Script的目录路径 script_dir = os.path.dirname(os.path.abspath(__file__)) # Script directory language_dir = "zh" if language == "Chinese" else "en" - data_file = os.path.join(script_dir, "../assets", language_dir, "scenery_datas.jsonl") + data_file = os.path.join(script_dir, "..", "assets", language_dir, "scenery_datas.jsonl") build = BuildPrompt(data_file) scenery_prompt = build.generate_prompt(seed, styles, artists)