Spaces:
Runtime error
Runtime error
File size: 2,162 Bytes
f2dbf59 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 |
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}")
|