



# coding=utf-8
# Copyright 2023 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 platform
import sys


a_						:     str	     =		'3'

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

print('OS platform:', platform.platform())
print('OS architecture:', platform.machine())

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())
except ImportError:
     print('Torch version:', None)

try:
     import transformers

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




import unittest

import numpy as np
import torch

from diffusers import KarrasVePipeline, KarrasVeScheduler, UNetaDModel
from diffusers.utils.testing_utils import enable_full_determinism, require_torch, slow, torch_device


enable_full_determinism()





class    lowerCamelCase__  (						unittest.TestCase):
     """simple docstring"""



     @property
     def 						_a      (self    ):






           '''simple docstring'''



           torch.manual_seed(0    )
           lowerCamelCase								=		UNetaDModel(
               block_out_channels=(32, 64)				,			layers_per_block=2				,			sample_size=32				,			in_channels=3				,			out_channels=3				,			down_block_types=("DownBlock2D", "AttnDownBlock2D")				,			up_block_types=("AttnUpBlock2D", "UpBlock2D")				,			)
           return model







     def 						_a      (self    ):






           '''simple docstring'''



           lowerCamelCase								=		self.dummy_uncond_unet
           lowerCamelCase								=		KarrasVeScheduler()

           lowerCamelCase								=		KarrasVePipeline(unet=__a				,			scheduler=__a    )
           pipe.to(__a    )
           pipe.set_progress_bar_config(disable=__a    )

           lowerCamelCase								=		torch.manual_seed(0    )
           lowerCamelCase								=		pipe(num_inference_steps=2				,			generator=__a				,			output_type="numpy"    ).images

           lowerCamelCase								=		torch.manual_seed(0    )
           lowerCamelCase								=		pipe(num_inference_steps=2				,			generator=__a				,			output_type="numpy"				,			return_dict=__a    )[0]

           lowerCamelCase								=		image[0, -3:, -3:, -1]
           lowerCamelCase								=		image_from_tuple[0, -3:, -3:, -1]

           assert image.shape == (1, 32, 32, 3)
           lowerCamelCase								=		np.array([0.0, 1.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0]    )

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





@slow
@require_torch
class    lowerCamelCase__  (						unittest.TestCase):
     """simple docstring"""



     def 						_a      (self    ):






           '''simple docstring'''



           lowerCamelCase								=		"google/ncsnpp-celebahq-256"
           lowerCamelCase								=		UNetaDModel.from_pretrained(__a    )
           lowerCamelCase								=		KarrasVeScheduler()

           lowerCamelCase								=		KarrasVePipeline(unet=__a				,			scheduler=__a    )
           pipe.to(__a    )
           pipe.set_progress_bar_config(disable=__a    )

           lowerCamelCase								=		torch.manual_seed(0    )
           lowerCamelCase								=		pipe(num_inference_steps=20				,			generator=__a				,			output_type="numpy"    ).images

           lowerCamelCase								=		image[0, -3:, -3:, -1]
           assert image.shape == (1, 2_56, 2_56, 3)
           lowerCamelCase								=		np.array([0.578, 0.5811, 0.5924, 0.5809, 0.587, 0.5886, 0.5861, 0.5802, 0.586]    )

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