jhj0517
Add temp dir
3fcd292
raw
history blame
523 Bytes
import os
WEBUI_DIR = os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
MODELS_DIR = os.path.join(WEBUI_DIR, "models")
SAM2_CONFIGS_DIR = os.path.join(WEBUI_DIR, "configs")
OUTPUT_DIR = os.path.join(WEBUI_DIR, "outputs")
OUTPUT_PSD_DIR = os.path.join(OUTPUT_DIR, "psd")
TEMP_DIR = os.path.join(WEBUI_DIR, "temp")
for dir_path in [MODELS_DIR,
SAM2_CONFIGS_DIR,
OUTPUT_DIR,
OUTPUT_PSD_DIR,
TEMP_DIR]:
os.makedirs(dir_path, exist_ok=True)