Spaces:
Runtime error
Runtime error
import os | |
import torch | |
import yaml | |
import comfy.utils | |
import numpy as np | |
import tempfile | |
from pathlib import Path | |
USE_SYMLINKS = False | |
here = Path(__file__).parent.resolve() | |
config_path = Path(here, "config.yaml") | |
ANNOTATOR_CKPTS_PATH = "" | |
TEMP_DIR = "" | |
USE_SYMLINKS = False | |
ORT_PROVIDERS = ["CUDAExecutionProvider", "DirectMLExecutionProvider", "OpenVINOExecutionProvider", "ROCMExecutionProvider", "CPUExecutionProvider", "CoreMLExecutionProvider"] | |
print(f'here: {here}') | |
if os.path.exists(config_path): | |
config = yaml.load(open(config_path, "r"), Loader=yaml.FullLoader) | |
ANNOTATOR_CKPTS_PATH = str(Path(here, config["annotator_ckpts_path"])) | |
TEMP_DIR = str(Path(here, config["custom_temp_path"]).resolve()) | |
USE_SYMLINKS = config["USE_SYMLINKS"] | |
ORT_PROVIDERS = config["EP_list"] | |
if TEMP_DIR is None: | |
TEMP_DIR = tempfile.gettempdir() | |
elif not os.path.isdir(TEMP_DIR): | |
try: | |
os.makedirs(TEMP_DIR) | |
except: | |
print(f"Failed to create custom temp directory. Using default.") | |
TEMP_DIR = tempfile.gettempdir() | |
if not os.path.isdir(ANNOTATOR_CKPTS_PATH): | |
try: | |
os.makedirs(ANNOTATOR_CKPTS_PATH) | |
except: | |
print(f"Failed to create config ckpts directory. Using default.") | |
ANNOTATOR_CKPTS_PATH = str(Path(here, "./ckpts")) | |
else: | |
ANNOTATOR_CKPTS_PATH = str(Path(here, "./ckpts")) | |
TEMP_DIR = tempfile.gettempdir() | |
USE_SYMLINKS = False | |
ORT_PROVIDERS = ["CUDAExecutionProvider", "DirectMLExecutionProvider", "OpenVINOExecutionProvider", "ROCMExecutionProvider", "CPUExecutionProvider", "CoreMLExecutionProvider"] | |
os.environ['AUX_ANNOTATOR_CKPTS_PATH'] = os.getenv('AUX_ANNOTATOR_CKPTS_PATH', ANNOTATOR_CKPTS_PATH) | |
os.environ['AUX_TEMP_DIR'] = os.getenv('AUX_TEMP_DIR', str(TEMP_DIR)) | |
os.environ['AUX_USE_SYMLINKS'] = os.getenv('AUX_USE_SYMLINKS', str(USE_SYMLINKS)) | |
os.environ['AUX_ORT_PROVIDERS'] = os.getenv('AUX_ORT_PROVIDERS', str(",".join(ORT_PROVIDERS))) | |
print(f"Using ckpts path: {ANNOTATOR_CKPTS_PATH}") | |
print(f"Using symlinks: {USE_SYMLINKS}") | |
print(f"Using ort providers: {ORT_PROVIDERS}") | |