


from math import sqrt






def 							_lowercase    (		SCREAMING_SNAKE_CASE_   :		int	):





			"""simple docstring"""




			if 1 < number < 4:
						# 2 and 3 are primes
						return True
			elif number < 2 or number % 2 == 0 or number % 3 == 0:
						# Negatives, 0, 1, all even numbers, all multiples of 3 are not primes
						return False

			# All primes number are in format of 6k +/- 1
			for i in range(5       ,  int(sqrt(SCREAMING_SNAKE_CASE_	) + 1	)       ,  6	):
						if number % i == 0 or number % (i + 2) == 0:
									return False
			return True






def 							_lowercase    (		SCREAMING_SNAKE_CASE_   :		int = 10_001	):





			"""simple docstring"""




			UpperCamelCase         =					0
			UpperCamelCase         =					1
			while count != nth and number < 3:
						number += 1
						if is_prime(SCREAMING_SNAKE_CASE_	):
									count += 1
			while count != nth:
						number += 2
						if is_prime(SCREAMING_SNAKE_CASE_	):
									count += 1
			return number


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








from ...utils import is_torch_available, is_transformers_available


if is_transformers_available() and is_torch_available():
			from .pipeline_vq_diffusion import LearnedClassifierFreeSamplingEmbeddings, VQDiffusionPipeline




