INNER CODE UNIT · Python
find_config_path
dusty-nv/jetson-containers · packages/vlm/vila-microservice/src/main.py:37
def find_config_path(base_env_var):
config_path = os.environ[base_env_var]
config_path_under_data_dir = os.environ[f"{base_env_var}_UNDER_DATA_DIR"]
if os.path.exists(config_path_under_data_dir):
return config_path_under_data_dir
else:
os.makedirs(os.path.dirname(config_path_under_data_dir), exist_ok=True)
shutil.copy(config_path, config_path_under_data_dir)
return config_path_under_data_dir
#Load config
config_path = find_config_path("MAIN_CONFIG_PATH")
config = load_config(config_path, "main")
logging.basicConfig(level=logging.getLevelName(config.log_level),
format='%(asctime)s - %(levelname)s - [%(threadName)s] %(message)s',
datefmt='%Y-%m-%d %H:%M:%S')