File size: 912 Bytes
a9b9771 b768d73 a9b9771 |
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 |
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 |