import onnxruntime as ort

VAE_CONFIGS = 'onnx_1/vae_decoder/config.json'

UNET_CONFIGS = 'onnx_1/unet/config.json'

TEXT_ENCODER_2_CONFIGS = 'onnx_1/text_encoder_2/config.json'

VAE_ONNX_PATH = 'onnx_1/vae_decoder/model.onnx'

UNET_ONNX_PATH = 'onnx_1/unet/canny_2/model.onnx'

TOKENIZER_PATH = 'onnx_1/tokenizer'

TOKENIZER_PATH2 = 'onnx_1/tokenizer_2'

TEXT_ENCODER_PATH = 'onnx_1/text_encoder/model.onnx'

TEXT_ENCODER_PATH2 = 'onnx_1/text_encoder_2/model.onnx'

SCHEDULER_PATH = 'onnx_1/scheduler'

CONTROLNEXT_ONNX_PATH = 'onnx_1/controlnext/model.onnx'

IMAGE_ENCODER_ONNX_PATH = 'onnx_1/image_encoder/model.onnx'

PROJ_ONNX_PATH = 'onnx_1/proj/model.onnx'

providers = ['CUDAExecutionProvider']

provider_options_0 = [{'device_id': 0}]

provider_options_1 = [{'device_id': 1}]

session_options = ort.SessionOptions()
session_options.enable_mem_pattern = False
session_options.enable_cpu_mem_arena = False