






'''simple docstring'''





def __UpperCAmelCase (			A			:	Optional[Any] = 2_0						)  -> Dict:







			try:
						UpperCAmelCase_     :   Optional[int]       =   int(lowercase_						)
			except (TypeError, ValueError):
						raise TypeError('''Parameter n must be int or castable to int.'''						)
			if n <= 0:
						raise ValueError('''Parameter n must be greater than or equal to one.'''						)
			UpperCAmelCase_     :   Tuple       =   0
			while 1:
						i += n * (n - 1)
						UpperCAmelCase_     :   Any       =   0
						for j in range(2	,   lowercase_						):
									if i % j != 0:
												UpperCAmelCase_     :   int       =   1
												break
						if nfound == 0:
									if i == 0:
												UpperCAmelCase_     :   List[Any]       =   1
									return i
			return None


if __name__ == "__main__":
						print(f'''{solution() = }''')














'''simple docstring'''





import unittest

import torch

from diffusers import DDIMScheduler, TextToVideoZeroPipeline
from diffusers.utils import load_pt, require_torch_gpu, slow

from ..test_pipelines_common import assert_mean_pixel_difference


@slow
@require_torch_gpu
class 				snake_case__     (			unittest.TestCase):




					def 						A     (       self       :						Optional[Any]     )  ->    List[str]:
								UpperCAmelCase_     :   Optional[Any]       =   '''runwayml/stable-diffusion-v1-5'''
								UpperCAmelCase_     :   Optional[int]       =   TextToVideoZeroPipeline.from_pretrained(_A							,				torch_dtype=torch.floataa     ).to('''cuda'''     )
								UpperCAmelCase_     :   Tuple       =   DDIMScheduler.from_config(pipe.scheduler.config     )
								UpperCAmelCase_     :   Tuple       =   torch.Generator(device='''cuda'''     ).manual_seed(0     )

								UpperCAmelCase_     :   List[Any]       =   '''A bear is playing a guitar on Times Square'''
								UpperCAmelCase_     :   List[Any]       =   pipe(prompt=_A							,				generator=_A     ).images

								UpperCAmelCase_     :   Any       =   load_pt(
								    '''https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/text-to-video/A bear is playing a guitar on Times Square.pt'''     )

								assert_mean_pixel_difference(_A							,				_A     )


