

import json
import os
from functools import lru_cache
from typing import TYPE_CHECKING, List, Optional, Tuple

import regex as re

from ...tokenization_utils import AddedToken, PreTrainedTokenizer
from ...utils import logging


if TYPE_CHECKING:
    from transformers.pipelines.conversational import Conversation

__A			:					Any						=			logging.get_logger(__name__)


__A			:					Any						=			{
    'vocab_file': 'vocab.json',
    'merges_file': 'merges.txt',
    'tokenizer_config_file': 'tokenizer_config.json',
}

__A			:					Optional[Any]						=			{
    'vocab_file': {'facebook/blenderbot-3B': 'https://huggingface.co/facebook/blenderbot-3B/resolve/main/vocab.json'},
    'merges_file': {'facebook/blenderbot-3B': 'https://huggingface.co/facebook/blenderbot-3B/resolve/main/merges.txt'},
    'tokenizer_config_file': {
        'facebook/blenderbot-3B': 'https://huggingface.co/facebook/blenderbot-3B/resolve/main/tokenizer_config.json'
    },
}

__A			:					Union[str, Any]						=			{'facebook/blenderbot-3B': 1_2_8}





@lru_cache()
# Copied from transformers.models.roberta.tokenization_roberta.bytes_to_unicode
def 	__a     (      ):
    SCREAMING_SNAKE_CASE										=   (
        list(range(ord("!"       )			,  ord("~"       ) + 1       )       ) + list(range(ord("¡"       )			,  ord("¬"       ) + 1       )       ) + list(range(ord("®"       )			,  ord("ÿ"       ) + 1       )       )
    )
    SCREAMING_SNAKE_CASE										=   bs[:]
    SCREAMING_SNAKE_CASE										=   0
    for b in range(2**8       ):
        if b not in bs:
            bs.append(A__       )
            cs.append(2**8 + n       )
            n += 1
    SCREAMING_SNAKE_CASE										=   [chr(A__       ) for n in cs]
    return dict(zip(A__			,  A__       )       )





def 	__a     (      A__			:  List[Any]       ):
    SCREAMING_SNAKE_CASE										=   set()
    SCREAMING_SNAKE_CASE										=   word[0]
    for char in word[1:]:
        pairs.add((prev_char, char)       )
        SCREAMING_SNAKE_CASE										=   char
    return pairs





class 			_SCREAMING_SNAKE_CASE (   __snake_case		):







   '''simple docstring'''




   lowerCamelCase__				=      VOCAB_FILES_NAMES
   lowerCamelCase__				=      PRETRAINED_VOCAB_FILES_MAP
   lowerCamelCase__				=      PRETRAINED_POSITIONAL_EMBEDDINGS_SIZES
   lowerCamelCase__				=      ["input_ids", "attention_mask"]





   def __init__(       self					:			Union[str, Any]     , __lowerCamelCase					:			Dict     , __lowerCamelCase					:			Optional[Any]     , __lowerCamelCase					:			Any="replace"     , __lowerCamelCase					:			Optional[Any]="<s>"     , __lowerCamelCase					:			Optional[Any]="</s>"     , __lowerCamelCase					:			Any="</s>"     , __lowerCamelCase					:			Union[str, Any]="<s>"     , __lowerCamelCase					:			List[str]="<unk>"     , __lowerCamelCase					:			Optional[Any]="<pad>"     , __lowerCamelCase					:			Dict="<mask>"     , __lowerCamelCase					:			Any=False     , **__lowerCamelCase					:			Optional[Any]     , ):
       SCREAMING_SNAKE_CASE										=   AddedToken(__lowerCamelCase     , lstrip=__lowerCamelCase     , rstrip=__lowerCamelCase			) if isinstance(__lowerCamelCase     , __lowerCamelCase			) else bos_token
       SCREAMING_SNAKE_CASE										=   AddedToken(__lowerCamelCase     , lstrip=__lowerCamelCase     , rstrip=__lowerCamelCase			) if isinstance(__lowerCamelCase     , __lowerCamelCase			) else eos_token
       SCREAMING_SNAKE_CASE										=   AddedToken(__lowerCamelCase     , lstrip=__lowerCamelCase     , rstrip=__lowerCamelCase			) if isinstance(__lowerCamelCase     , __lowerCamelCase			) else sep_token
       SCREAMING_SNAKE_CASE										=   AddedToken(__lowerCamelCase     , lstrip=__lowerCamelCase     , rstrip=__lowerCamelCase			) if isinstance(__lowerCamelCase     , __lowerCamelCase			) else cls_token
       SCREAMING_SNAKE_CASE										=   AddedToken(__lowerCamelCase     , lstrip=__lowerCamelCase     , rstrip=__lowerCamelCase			) if isinstance(__lowerCamelCase     , __lowerCamelCase			) else unk_token
       SCREAMING_SNAKE_CASE										=   AddedToken(__lowerCamelCase     , lstrip=__lowerCamelCase     , rstrip=__lowerCamelCase			) if isinstance(__lowerCamelCase     , __lowerCamelCase			) else pad_token

       # Mask token behave like a normal word, i.e. include the space before it
       SCREAMING_SNAKE_CASE										=   AddedToken(__lowerCamelCase     , lstrip=__lowerCamelCase     , rstrip=__lowerCamelCase			) if isinstance(__lowerCamelCase     , __lowerCamelCase			) else mask_token

       super().__init__(
           errors=__lowerCamelCase     , bos_token=__lowerCamelCase     , eos_token=__lowerCamelCase     , unk_token=__lowerCamelCase     , sep_token=__lowerCamelCase     , cls_token=__lowerCamelCase     , pad_token=__lowerCamelCase     , mask_token=__lowerCamelCase     , add_prefix_space=__lowerCamelCase     , **__lowerCamelCase     , )

       with open(__lowerCamelCase     , encoding="utf-8"			) as vocab_handle:
           SCREAMING_SNAKE_CASE										=   json.load(__lowerCamelCase			)
       SCREAMING_SNAKE_CASE										=   {v: k for k, v in self.encoder.items()}
       SCREAMING_SNAKE_CASE										=   errors  # how to handle errors in decoding
       SCREAMING_SNAKE_CASE										=   bytes_to_unicode()
       SCREAMING_SNAKE_CASE										=   {v: k for k, v in self.byte_encoder.items()}
       with open(__lowerCamelCase     , encoding="utf-8"			) as merges_handle:
           SCREAMING_SNAKE_CASE										=   merges_handle.read().split("\n"			)[1:-1]
       SCREAMING_SNAKE_CASE										=   [tuple(merge.split()			) for merge in bpe_merges]
       SCREAMING_SNAKE_CASE										=   dict(zip(__lowerCamelCase     , range(len(__lowerCamelCase			)			)			)			)
       SCREAMING_SNAKE_CASE										=   {}
       SCREAMING_SNAKE_CASE										=   add_prefix_space

       # Should have added re.IGNORECASE so BPE merges can happen for capitalized versions of contractions
       SCREAMING_SNAKE_CASE										=   re.compile(r"'s|'t|'re|'ve|'m|'ll|'d| ?\p{L}+| ?\p{N}+| ?[^\s\p{L}\p{N}]+|\s+(?!\S)|\s+"			)





   @property
   # Copied from transformers.models.roberta.tokenization_roberta.RobertaTokenizer.vocab_size with Roberta->Blenderbot, RoBERTa->Blenderbot
   def      _snake_case		(       self					:			str			):
       return len(self.encoder			)





   def      _snake_case		(       self					:			Union[str, Any]			):
       return dict(self.encoder     , **self.added_tokens_encoder			)





   def      _snake_case		(       self					:			Dict     , __lowerCamelCase					:			List[Any]			):
       if token in self.cache:
           return self.cache[token]
       SCREAMING_SNAKE_CASE										=   tuple(__lowerCamelCase			)
       SCREAMING_SNAKE_CASE										=   get_pairs(__lowerCamelCase			)

       if not pairs:
           return token

       while True:
           SCREAMING_SNAKE_CASE										=   min(__lowerCamelCase     , key=lambda __lowerCamelCase			: self.bpe_ranks.get(__lowerCamelCase     , float("inf"			)			)			)
           if bigram not in self.bpe_ranks:
               break
           SCREAMING_SNAKE_CASE							,	SCREAMING_SNAKE_CASE										=   bigram
           SCREAMING_SNAKE_CASE										=   []
           SCREAMING_SNAKE_CASE										=   0
           while i < len(__lowerCamelCase			):
               try:
                   SCREAMING_SNAKE_CASE										=   word.index(__lowerCamelCase     , __lowerCamelCase			)
               except ValueError:
                   new_word.extend(word[i:]			)
                   break
               else:
                   new_word.extend(word[i:j]			)
                   SCREAMING_SNAKE_CASE										=   j

               if word[i] == first and i < len(__lowerCamelCase			) - 1 and word[i + 1] == second:
                   new_word.append(first + second			)
                   i += 2
               else:
                   new_word.append(word[i]			)
                   i += 1
           SCREAMING_SNAKE_CASE										=   tuple(__lowerCamelCase			)
           SCREAMING_SNAKE_CASE										=   new_word
           if len(__lowerCamelCase			) == 1:
               break
           else:
               SCREAMING_SNAKE_CASE										=   get_pairs(__lowerCamelCase			)
       SCREAMING_SNAKE_CASE										=   " ".join(__lowerCamelCase			)
       SCREAMING_SNAKE_CASE										=   word
       return word





   def      _snake_case		(       self					:			Optional[Any]     , __lowerCamelCase					:			List[Any]			):
       SCREAMING_SNAKE_CASE										=   []
       for token in re.findall(self.pat     , __lowerCamelCase			):
           SCREAMING_SNAKE_CASE										=   "".join(
               self.byte_encoder[b] for b in token.encode("utf-8"			)			)  # Maps all our bytes to unicode strings, avoiding control tokens of the BPE (spaces in our case)
           bpe_tokens.extend(bpe_token for bpe_token in self.bpe(__lowerCamelCase			).split(" "			)			)
       return bpe_tokens





   def      _snake_case		(       self					:			Tuple     , __lowerCamelCase					:			Dict			):
       return self.encoder.get(__lowerCamelCase     , self.encoder.get(self.unk_token			)			)





   def      _snake_case		(       self					:			Any     , __lowerCamelCase					:			Optional[int]			):
       return self.decoder.get(__lowerCamelCase			)





   def      _snake_case		(       self					:			Optional[int]     , __lowerCamelCase					:			List[Any]			):
       SCREAMING_SNAKE_CASE										=   "".join(__lowerCamelCase			)
       SCREAMING_SNAKE_CASE										=   bytearray([self.byte_decoder[c] for c in text]			).decode("utf-8"     , errors=self.errors			)
       return text





   def      _snake_case		(       self					:			Any     , __lowerCamelCase					:			str     , __lowerCamelCase					:			Optional[str] = None			):
       if not os.path.isdir(__lowerCamelCase			):
           logger.error(f"Vocabulary path ({save_directory}) should be a directory"			)
           return
       SCREAMING_SNAKE_CASE										=   os.path.join(
           __lowerCamelCase     , (filename_prefix + "-" if filename_prefix else "") + VOCAB_FILES_NAMES["vocab_file"]			)
       SCREAMING_SNAKE_CASE										=   os.path.join(
           __lowerCamelCase     , (filename_prefix + "-" if filename_prefix else "") + VOCAB_FILES_NAMES["merges_file"]			)

       with open(__lowerCamelCase     , "w"     , encoding="utf-8"			) as f:
           f.write(json.dumps(self.encoder     , indent=2     , sort_keys=__lowerCamelCase     , ensure_ascii=__lowerCamelCase			) + "\n"			)

       SCREAMING_SNAKE_CASE										=   0
       with open(__lowerCamelCase     , "w"     , encoding="utf-8"			) as writer:
           writer.write("#version: 0.2\n"			)
           for bpe_tokens, token_index in sorted(self.bpe_ranks.items()     , key=lambda __lowerCamelCase			: kv[1]			):
               if index != token_index:
                   logger.warning(
                       f"Saving vocabulary to {merge_file}: BPE merge indices are not consecutive."
                       " Please check that the tokenizer is not corrupted!"			)
                   SCREAMING_SNAKE_CASE										=   token_index
               writer.write(" ".join(__lowerCamelCase			) + "\n"			)
               index += 1

       return vocab_file, merge_file





   def      _snake_case		(       self					:			Dict     , __lowerCamelCase					:			List[int]     , __lowerCamelCase					:			Optional[List[int]] = None     , __lowerCamelCase					:			bool = False			):
       if already_has_special_tokens:
           return super().get_special_tokens_mask(
               token_ids_a=__lowerCamelCase     , token_ids_a=__lowerCamelCase     , already_has_special_tokens=__lowerCamelCase			)

       if token_ids_a is None:
           return [1] + ([0] * len(__lowerCamelCase			)) + [1]
       return [1] + ([0] * len(__lowerCamelCase			)) + [1, 1] + ([0] * len(__lowerCamelCase			)) + [1]





   def      _snake_case		(       self					:			Optional[Any]     , __lowerCamelCase					:			List[int]     , __lowerCamelCase					:			Optional[List[int]] = None			):
       SCREAMING_SNAKE_CASE										=   [self.sep_token_id]
       SCREAMING_SNAKE_CASE										=   [self.cls_token_id]

       if token_ids_a is None:
           return len(cls + token_ids_a + sep			) * [0]
       return len(cls + token_ids_a + sep + sep + token_ids_a + sep			) * [0]





   def      _snake_case		(       self					:			Optional[int]     , __lowerCamelCase					:			Dict     , __lowerCamelCase					:			Any=False     , **__lowerCamelCase					:			Union[str, Any]			):
       SCREAMING_SNAKE_CASE										=   kwargs.pop("add_prefix_space"     , self.add_prefix_space			)
       if (is_split_into_words or add_prefix_space) and (len(__lowerCamelCase			) > 0 and not text[0].isspace()):
           SCREAMING_SNAKE_CASE										=   " " + text
       return (text, kwargs)





   def      _snake_case		(       self					:			Any     , __lowerCamelCase					:			List[int]     , __lowerCamelCase					:			Optional[List[int]] = None			):
       return token_ids_a + [self.eos_token_id]





   def      _snake_case		(       self					:			int     , __lowerCamelCase					:			"Conversation"			):
       SCREAMING_SNAKE_CASE										=   []
       for is_user, text in conversation.iter_texts():
           if is_user:
               # We need to space prefix as it's being done within blenderbot
               inputs.append(" " + text			)
           else:
               # Generated responses should contain them already.
               inputs.append(__lowerCamelCase			)

       SCREAMING_SNAKE_CASE										=   "  ".join(__lowerCamelCase			)
       SCREAMING_SNAKE_CASE										=   self.encode(__lowerCamelCase			)
       if len(__lowerCamelCase			) > self.model_max_length:
           SCREAMING_SNAKE_CASE										=   input_ids[-self.model_max_length :]
           logger.warning(f"Trimmed input from conversation as it was longer than {self.model_max_length} tokens."			)
       return input_ids

"""simple docstring"""


import random
import unittest

import numpy as np

from diffusers import (
    DPMSolverMultistepScheduler,
    EulerAncestralDiscreteScheduler,
    EulerDiscreteScheduler,
    LMSDiscreteScheduler,
    OnnxStableDiffusionImgaImgPipeline,
    PNDMScheduler,
)
from diffusers.utils import floats_tensor
from diffusers.utils.testing_utils import (
    is_onnx_available,
    load_image,
    nightly,
    require_onnxruntime,
    require_torch_gpu,
)

from ..test_pipelines_onnx_common import OnnxPipelineTesterMixin


if is_onnx_available():
		import onnxruntime as ort







class 					lowerCAmelCase_					(   _lowercase		,       unittest.TestCase     ):



						'''simple docstring'''


						_lowerCamelCase:     Dict   =     '''hf-internal-testing/tiny-random-OnnxStableDiffusionPipeline'''


						def     _SCREAMING_SNAKE_CASE   (			self  :   Any	,A_  :   List[str]=0   )  -> str:
							A     = floats_tensor((1, 3, 128, 128)	,rng=random.Random(A_   )   )
							A     = np.random.RandomState(A_   )
							A     = {
							    'prompt': 'A painting of a squirrel eating a burger',
							    'image': image,
							    'generator': generator,
							    'num_inference_steps': 3,
							    'strength': 0.75,
							    'guidance_scale': 7.5,
							    'output_type': 'numpy',
							}
							return inputs


						def     _SCREAMING_SNAKE_CASE   (			self  :   Dict   )  -> Tuple:
							A     = OnnxStableDiffusionImgaImgPipeline.from_pretrained(self.hub_checkpoint	,provider='CPUExecutionProvider'   )
							pipe.set_progress_bar_config(disable=A_   )

							A     = self.get_dummy_inputs()
							A     = pipe(**A_   ).images
							A     = image[0, -3:, -3:, -1].flatten()

							assert image.shape == (1, 128, 128, 3)
							A     = np.array([0.6_96_43, 0.5_84_84, 0.5_03_14, 0.5_87_60, 0.5_53_68, 0.5_96_43, 0.5_15_29, 0.4_12_17, 0.4_90_87]   )
							assert np.abs(image_slice - expected_slice   ).max() < 1e-1


						def     _SCREAMING_SNAKE_CASE   (			self  :   List[Any]   )  -> Any:
							A     = OnnxStableDiffusionImgaImgPipeline.from_pretrained(self.hub_checkpoint	,provider='CPUExecutionProvider'   )
							A     = PNDMScheduler.from_config(pipe.scheduler.config	,skip_prk_steps=A_   )
							pipe.set_progress_bar_config(disable=A_   )

							A     = self.get_dummy_inputs()
							A     = pipe(**A_   ).images
							A     = image[0, -3:, -3:, -1]

							assert image.shape == (1, 128, 128, 3)
							A     = np.array([0.6_17_37, 0.5_46_42, 0.5_31_83, 0.5_44_65, 0.5_27_42, 0.6_05_25, 0.4_99_69, 0.4_06_55, 0.4_81_54]   )

							assert np.abs(image_slice.flatten() - expected_slice   ).max() < 1e-1


						def     _SCREAMING_SNAKE_CASE   (			self  :   Union[str, Any]   )  -> int:
							A     = OnnxStableDiffusionImgaImgPipeline.from_pretrained(self.hub_checkpoint	,provider='CPUExecutionProvider'   )
							A     = LMSDiscreteScheduler.from_config(pipe.scheduler.config   )
							pipe.set_progress_bar_config(disable=A_   )

							# warmup pass to apply optimizations
							A     = pipe(**self.get_dummy_inputs()   )

							A     = self.get_dummy_inputs()
							A     = pipe(**A_   ).images
							A     = image[0, -3:, -3:, -1]

							assert image.shape == (1, 128, 128, 3)
							A     = np.array([0.5_27_61, 0.5_99_77, 0.4_90_33, 0.4_96_19, 0.5_42_82, 0.5_03_11, 0.4_76_00, 0.4_09_18, 0.4_52_03]   )

							assert np.abs(image_slice.flatten() - expected_slice   ).max() < 1e-1


						def     _SCREAMING_SNAKE_CASE   (			self  :   Optional[int]   )  -> Tuple:
							A     = OnnxStableDiffusionImgaImgPipeline.from_pretrained(self.hub_checkpoint	,provider='CPUExecutionProvider'   )
							A     = EulerDiscreteScheduler.from_config(pipe.scheduler.config   )
							pipe.set_progress_bar_config(disable=A_   )

							A     = self.get_dummy_inputs()
							A     = pipe(**A_   ).images
							A     = image[0, -3:, -3:, -1]

							assert image.shape == (1, 128, 128, 3)
							A     = np.array([0.5_29_11, 0.6_00_04, 0.4_92_29, 0.4_98_05, 0.5_45_02, 0.5_06_80, 0.4_77_77, 0.4_10_28, 0.4_53_04]   )

							assert np.abs(image_slice.flatten() - expected_slice   ).max() < 1e-1


						def     _SCREAMING_SNAKE_CASE   (			self  :   Optional[Any]   )  -> str:
							A     = OnnxStableDiffusionImgaImgPipeline.from_pretrained(self.hub_checkpoint	,provider='CPUExecutionProvider'   )
							A     = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config   )
							pipe.set_progress_bar_config(disable=A_   )

							A     = self.get_dummy_inputs()
							A     = pipe(**A_   ).images
							A     = image[0, -3:, -3:, -1]

							assert image.shape == (1, 128, 128, 3)
							A     = np.array([0.5_29_11, 0.6_00_04, 0.4_92_29, 0.4_98_05, 0.5_45_02, 0.5_06_80, 0.4_77_77, 0.4_10_28, 0.4_53_04]   )

							assert np.abs(image_slice.flatten() - expected_slice   ).max() < 1e-1




						def     _SCREAMING_SNAKE_CASE   (			self  :   int   )  -> Tuple:
							A     = OnnxStableDiffusionImgaImgPipeline.from_pretrained(self.hub_checkpoint	,provider='CPUExecutionProvider'   )
							A     = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config   )
							pipe.set_progress_bar_config(disable=A_   )

							A     = self.get_dummy_inputs()
							A     = pipe(**A_   ).images
							A     = image[0, -3:, -3:, -1]

							assert image.shape == (1, 128, 128, 3)
							A     = np.array([0.6_53_31, 0.5_82_77, 0.4_82_04, 0.5_60_59, 0.5_36_65, 0.5_62_35, 0.5_09_69, 0.4_00_09, 0.4_65_52]   )

							assert np.abs(image_slice.flatten() - expected_slice   ).max() < 1e-1







@nightly
@require_onnxruntime
@require_torch_gpu
class 					lowerCAmelCase_					(   unittest.TestCase     ):



						'''simple docstring'''


						@property
						def     _SCREAMING_SNAKE_CASE   (			self  :   int   )  -> Union[str, Any]:
							return (
							    "CUDAExecutionProvider",
							    {
							        "gpu_mem_limit": "15000000000",  # 15GB
							        "arena_extend_strategy": "kSameAsRequested",
							    },
							)


						@property
						def     _SCREAMING_SNAKE_CASE   (			self  :   str   )  -> List[Any]:
							A     = ort.SessionOptions()
							A     = False
							return options


						def     _SCREAMING_SNAKE_CASE   (			self  :   Tuple   )  -> str:
							A     = load_image(
							    'https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main'
							    '/img2img/sketch-mountains-input.jpg'   )
							A     = init_image.resize((768, 512)   )
							# using the PNDM scheduler by default
							A     = OnnxStableDiffusionImgaImgPipeline.from_pretrained(
							    'CompVis/stable-diffusion-v1-4'	,revision='onnx'	,safety_checker=A_	,feature_extractor=A_	,provider=self.gpu_provider	,sess_options=self.gpu_options	,)
							pipe.set_progress_bar_config(disable=A_   )

							A     = 'A fantasy landscape, trending on artstation'

							A     = np.random.RandomState(0   )
							A     = pipe(
							    prompt=A_	,image=A_	,strength=0.75	,guidance_scale=7.5	,num_inference_steps=10	,generator=A_	,output_type='np'	,)
							A     = output.images
							A     = images[0, 255:258, 383:386, -1]

							assert images.shape == (1, 512, 768, 3)
							A     = np.array([0.49_09, 0.50_59, 0.53_72, 0.46_23, 0.48_76, 0.50_49, 0.48_20, 0.49_56, 0.50_19]   )
							# TODO: lower the tolerance after finding the cause of onnxruntime reproducibility issues

							assert np.abs(image_slice.flatten() - expected_slice   ).max() < 2e-2




						def     _SCREAMING_SNAKE_CASE   (			self  :   Optional[int]   )  -> Optional[Any]:
							A     = load_image(
							    'https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main'
							    '/img2img/sketch-mountains-input.jpg'   )
							A     = init_image.resize((768, 512)   )
							A     = LMSDiscreteScheduler.from_pretrained(
							    'runwayml/stable-diffusion-v1-5'	,subfolder='scheduler'	,revision='onnx'   )
							A     = OnnxStableDiffusionImgaImgPipeline.from_pretrained(
							    'runwayml/stable-diffusion-v1-5'	,revision='onnx'	,scheduler=A_	,safety_checker=A_	,feature_extractor=A_	,provider=self.gpu_provider	,sess_options=self.gpu_options	,)
							pipe.set_progress_bar_config(disable=A_   )

							A     = 'A fantasy landscape, trending on artstation'

							A     = np.random.RandomState(0   )
							A     = pipe(
							    prompt=A_	,image=A_	,strength=0.75	,guidance_scale=7.5	,num_inference_steps=20	,generator=A_	,output_type='np'	,)
							A     = output.images
							A     = images[0, 255:258, 383:386, -1]

							assert images.shape == (1, 512, 768, 3)
							A     = np.array([0.80_43, 0.9_26, 0.95_81, 0.81_19, 0.89_54, 0.9_13, 0.72_09, 0.74_63, 0.74_31]   )
							# TODO: lower the tolerance after finding the cause of onnxruntime reproducibility issues

							assert np.abs(image_slice.flatten() - expected_slice   ).max() < 2e-2