diff --git a/Rayko_VAE_Save_Image.py b/Rayko_VAE_Save_Image.py index 1f525e9..7784fd2 100644 --- a/Rayko_VAE_Save_Image.py +++ b/Rayko_VAE_Save_Image.py @@ -79,7 +79,7 @@ class RS_VAE_Decode_Save: def _get_allowed_roots(self, node_data_raw: str = "") -> list: roots = {os.path.realpath(self.output_dir)} - env_raw = os.environ["RS_EXTRA_OUTPUT_ROOTS"] if "RS_EXTRA_OUTPUT_ROOTS" in os.environ else "" + env_raw = os.getenv("RS_EXTRA_OUTPUT_ROOTS", "") for part in env_raw.split(os.pathsep): part = part.strip() if part: