






"""simple docstring"""






def       _A    (__a							)				-> bool:
    """simple docstring"""






    SCREAMING_SNAKE_CASE_   :						Optional[int]							=	0
    for ch in input_str:
        SCREAMING_SNAKE_CASE_   :						Optional[int]							=	ord(__a							)
        SCREAMING_SNAKE_CASE_   :						Any							=	pow(2					,       __a							)

        # If we already turned on bit for current character's unicode
        if bitmap >> ch_unicode & 1 == 1:
            return False
        bitmap |= ch_bit_index_on
    return True


if __name__ == "__main__":
  import doctest

  doctest.testmod()










"""simple docstring"""






import unittest

import numpy as np
import torch

from diffusers import ScoreSdeVePipeline, ScoreSdeVeScheduler, 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 _SCREAMING_SNAKE_CASE				(		self  :						Tuple):




          '''simple docstring'''


          torch.manual_seed(0)
          SCREAMING_SNAKE_CASE_   :						Optional[Any]							=	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 _SCREAMING_SNAKE_CASE				(		self  :						Optional[Any]):




          '''simple docstring'''


          SCREAMING_SNAKE_CASE_   :						int							=	self.dummy_uncond_unet
          SCREAMING_SNAKE_CASE_   :						int							=	ScoreSdeVeScheduler()

          SCREAMING_SNAKE_CASE_   :						List[str]							=	ScoreSdeVePipeline(unet=lowercase_		,			scheduler=lowercase_)
          sde_ve.to(lowercase_)
          sde_ve.set_progress_bar_config(disable=lowercase_)

          SCREAMING_SNAKE_CASE_   :						Tuple							=	torch.manual_seed(0)
          SCREAMING_SNAKE_CASE_   :						Dict							=	sde_ve(num_inference_steps=2		,			output_type='''numpy'''		,			generator=lowercase_).images

          SCREAMING_SNAKE_CASE_   :						Optional[Any]							=	torch.manual_seed(0)
          SCREAMING_SNAKE_CASE_   :						Optional[int]							=	sde_ve(num_inference_steps=2		,			output_type='''numpy'''		,			generator=lowercase_		,			return_dict=lowercase_)[
              0
          ]

          SCREAMING_SNAKE_CASE_   :						Dict							=	image[0, -3:, -3:, -1]
          SCREAMING_SNAKE_CASE_   :						List[Any]							=	image_from_tuple[0, -3:, -3:, -1]

          assert image.shape == (1, 32, 32, 3)
          SCREAMING_SNAKE_CASE_   :						str							=	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 _SCREAMING_SNAKE_CASE				(		self  :						Dict):




          '''simple docstring'''


          SCREAMING_SNAKE_CASE_   :						Union[str, Any]							=	'''google/ncsnpp-church-256'''
          SCREAMING_SNAKE_CASE_   :						str							=	UNetaDModel.from_pretrained(lowercase_)

          SCREAMING_SNAKE_CASE_   :						Any							=	ScoreSdeVeScheduler.from_pretrained(lowercase_)

          SCREAMING_SNAKE_CASE_   :						Any							=	ScoreSdeVePipeline(unet=lowercase_		,			scheduler=lowercase_)
          sde_ve.to(lowercase_)
          sde_ve.set_progress_bar_config(disable=lowercase_)

          SCREAMING_SNAKE_CASE_   :						Optional[int]							=	torch.manual_seed(0)
          SCREAMING_SNAKE_CASE_   :						List[str]							=	sde_ve(num_inference_steps=10		,			output_type='''numpy'''		,			generator=lowercase_).images

          SCREAMING_SNAKE_CASE_   :						Any							=	image[0, -3:, -3:, -1]

          assert image.shape == (1, 256, 256, 3)

          SCREAMING_SNAKE_CASE_   :						int							=	np.array([0.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 0.0, 0.0])

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


