
# coding=utf-8
# Copyright 2020 The HuggingFace Inc. team.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

# this script dumps information about the environment

import os
import sys

import transformers


A     :	int    = '3'

print('Python version:', sys.version)
print('transformers version:', transformers.__version__)

try:
					import torch

					print('Torch version:', torch.__version__)
					print('Cuda available:', torch.cuda.is_available())
					print('Cuda version:', torch.version.cuda)
					print('CuDNN version:', torch.backends.cudnn.version())
					print('Number of GPUs available:', torch.cuda.device_count())
					print('NCCL version:', torch.cuda.nccl.version())
except ImportError:
					print('Torch version:', None)

try:
					import deepspeed

					print('DeepSpeed version:', deepspeed.__version__)
except ImportError:
					print('DeepSpeed version:', None)

try:
					import tensorflow as tf

					print('TensorFlow version:', tf.__version__)
					print('TF GPUs available:', bool(tf.config.list_physical_devices('GPU')))
					print('Number of TF GPUs available:', len(tf.config.list_physical_devices('GPU')))
except ImportError:
					print('TensorFlow version:', None)

import unittest

import numpy as np

from diffusers import LMSDiscreteScheduler, OnnxStableDiffusionInpaintPipeline
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 			__A( a  ,   unittest.TestCase			):
				# FIXME: add fast tests
				pass







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




				@property
				def 				SCREAMING_SNAKE_CASE_		(     self				)    ->      List[str]:


								'''simple docstring'''




								return (
								    "CUDAExecutionProvider",
								    {
								        "gpu_mem_limit": "15000000000",  # 15GB
								        "arena_extend_strategy": "kSameAsRequested",
								    },
								)




				@property
				def 				SCREAMING_SNAKE_CASE_		(     self				)    ->      Union[str, Any]:


								'''simple docstring'''




								__a										=    ort.SessionOptions()
								__a										=    False
								return options




				def 				SCREAMING_SNAKE_CASE_		(     self				)    ->      Optional[Any]:


								'''simple docstring'''




								__a										=    load_image(
								    '''https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main'''
								    '''/in_paint/overture-creations-5sI6fQgYIuo.png'''				)
								__a										=    load_image(
								    '''https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main'''
								    '''/in_paint/overture-creations-5sI6fQgYIuo_mask.png'''				)
								__a										=    OnnxStableDiffusionInpaintPipeline.from_pretrained(
								    '''runwayml/stable-diffusion-inpainting'''		,     revision='''onnx'''		,     safety_checker=_snake_case		,     feature_extractor=_snake_case		,     provider=self.gpu_provider		,     sess_options=self.gpu_options		,     )
								pipe.set_progress_bar_config(disable=_snake_case				)

								__a										=    '''A red cat sitting on a park bench'''

								__a										=    np.random.RandomState(0				)
								__a										=    pipe(
								    prompt=_snake_case		,     image=_snake_case		,     mask_image=_snake_case		,     guidance_scale=7.5		,     num_inference_steps=10		,     generator=_snake_case		,     output_type='''np'''		,     )
								__a										=    output.images
								__a										=    images[0, 255:258, 255:258, -1]

								assert images.shape == (1, 512, 512, 3)
								__a										=    np.array([0.2514, 0.3007, 0.3517, 0.1790, 0.2382, 0.3167, 0.1944, 0.2273, 0.2464]				)

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




				def 				SCREAMING_SNAKE_CASE_		(     self				)    ->      Tuple:


								'''simple docstring'''




								__a										=    load_image(
								    '''https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main'''
								    '''/in_paint/overture-creations-5sI6fQgYIuo.png'''				)
								__a										=    load_image(
								    '''https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main'''
								    '''/in_paint/overture-creations-5sI6fQgYIuo_mask.png'''				)
								__a										=    LMSDiscreteScheduler.from_pretrained(
								    '''runwayml/stable-diffusion-inpainting'''		,     subfolder='''scheduler'''		,     revision='''onnx'''				)
								__a										=    OnnxStableDiffusionInpaintPipeline.from_pretrained(
								    '''runwayml/stable-diffusion-inpainting'''		,     revision='''onnx'''		,     scheduler=_snake_case		,     safety_checker=_snake_case		,     feature_extractor=_snake_case		,     provider=self.gpu_provider		,     sess_options=self.gpu_options		,     )
								pipe.set_progress_bar_config(disable=_snake_case				)

								__a										=    '''A red cat sitting on a park bench'''

								__a										=    np.random.RandomState(0				)
								__a										=    pipe(
								    prompt=_snake_case		,     image=_snake_case		,     mask_image=_snake_case		,     guidance_scale=7.5		,     num_inference_steps=20		,     generator=_snake_case		,     output_type='''np'''		,     )
								__a										=    output.images
								__a										=    images[0, 255:258, 255:258, -1]

								assert images.shape == (1, 512, 512, 3)
								__a										=    np.array([0.0086, 0.0077, 0.0083, 0.0093, 0.0107, 0.0139, 0.0094, 0.0097, 0.0125]				)

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