





import argparse
import os
import re

import packaging.version


UpperCAmelCase_      							=     """examples/"""
UpperCAmelCase_      							=     {
    """examples""": (re.compile(R"""^check_min_version\(\"[^\"]+\"\)\s*$""", re.MULTILINE), """check_min_version(\"VERSION\")\n"""),
    """init""": (re.compile(R"""^__version__\s+=\s+\"([^\"]+)\"\s*$""", re.MULTILINE), """__version__ = \"VERSION\"\n"""),
    """setup""": (re.compile(R"""^(\s*)version\s*=\s*\"[^\"]+\",""", re.MULTILINE), R"""\1version=\"VERSION\","""),
    """doc""": (re.compile(R"""^(\s*)release\s*=\s*\"[^\"]+\"$""", re.MULTILINE), """release = \"VERSION\"\n"""),
}
UpperCAmelCase_      							=     {
    """init""": """src/diffusers/__init__.py""",
    """setup""": """setup.py""",
}
UpperCAmelCase_      							=     """README.md"""





def 						lowerCamelCase__     (     UpperCamelCase__			:       Tuple       ,    UpperCamelCase__			:       str       ,    UpperCamelCase__			:       Dict						)			-> Dict:



					'''simple docstring'''




					with open(UpperCamelCase__       ,    'r'       ,    encoding='utf-8'       ,    newline='\n'						) as f:
										_snake_case									=				f.read()
					_snake_case							,     _snake_case									=				REPLACE_PATTERNS[pattern]
					_snake_case									=				replace.replace('VERSION'       ,    UpperCamelCase__						)
					_snake_case									=				re_pattern.sub(UpperCamelCase__       ,    UpperCamelCase__						)
					with open(UpperCamelCase__       ,    'w'       ,    encoding='utf-8'       ,    newline='\n'						) as f:
										f.write(UpperCamelCase__						)





def 						lowerCamelCase__     (     UpperCamelCase__			:       Union[str, Any]						)			-> List[Any]:



					'''simple docstring'''




					for folder, directories, fnames in os.walk(UpperCamelCase__						):
										# Removing some of the folders with non-actively maintained examples from the walk
										if "research_projects" in directories:
															directories.remove('research_projects'						)
										if "legacy" in directories:
															directories.remove('legacy'						)
										for fname in fnames:
															if fname.endswith('.py'						):
																				update_version_in_file(os.path.join(UpperCamelCase__       ,    UpperCamelCase__						)       ,    UpperCamelCase__       ,    pattern='examples'						)





def 						lowerCamelCase__     (     UpperCamelCase__			:       Dict       ,    UpperCamelCase__			:       Dict=False						)			-> List[Any]:



					'''simple docstring'''




					for pattern, fname in REPLACE_FILES.items():
										update_version_in_file(UpperCamelCase__       ,    UpperCamelCase__       ,    UpperCamelCase__						)
					if not patch:
										update_version_in_examples(UpperCamelCase__						)





def 						lowerCamelCase__     (     )			-> str:



					'''simple docstring'''




					_snake_case									=				'🤗 Transformers currently provides the following architectures'
					_snake_case									=				'1. Want to contribute a new model?'
					with open(UpperCamelCase__       ,    'r'       ,    encoding='utf-8'       ,    newline='\n'						) as f:
										_snake_case									=				f.readlines()

					# Find the start of the list.
					_snake_case									=				0
					while not lines[start_index].startswith(_start_prompt						):
										start_index += 1
					start_index += 1

					_snake_case									=				start_index
					# Update the lines in the model list.
					while not lines[index].startswith(_end_prompt						):
										if lines[index].startswith('1.'						):
															_snake_case									=				lines[index].replace(
															    'https://huggingface.co/docs/diffusers/main/model_doc'       ,    'https://huggingface.co/docs/diffusers/model_doc'       ,    )
										index += 1

					with open(UpperCamelCase__       ,    'w'       ,    encoding='utf-8'       ,    newline='\n'						) as f:
										f.writelines(UpperCamelCase__						)





def 						lowerCamelCase__     (     )			-> int:



					'''simple docstring'''




					with open(REPLACE_FILES['init']       ,    'r'						) as f:
										_snake_case									=				f.read()
					_snake_case									=				REPLACE_PATTERNS['init'][0].search(UpperCamelCase__						).groups()[0]
					return packaging.version.parse(UpperCamelCase__						)





def 						lowerCamelCase__     (     UpperCamelCase__			:       int=False						)			-> Optional[int]:



					'''simple docstring'''




					_snake_case									=				get_version()
					if patch and default_version.is_devrelease:
										raise ValueError('Can\'t create a patch version from the dev branch, checkout a released version!'						)
					if default_version.is_devrelease:
										_snake_case									=				default_version.base_version
					elif patch:
										_snake_case									=				F'''{default_version.major}.{default_version.minor}.{default_version.micro + 1}'''
					else:
										_snake_case									=				F'''{default_version.major}.{default_version.minor + 1}.0'''

					# Now let's ask nicely if that's the right one.
					_snake_case									=				input(F'''Which version are you releasing? [{default_version}]'''						)
					if len(UpperCamelCase__						) == 0:
										_snake_case									=				default_version

					print(F'''Updating version to {version}.'''						)
					global_version_update(UpperCamelCase__       ,    patch=UpperCamelCase__						)





def 						lowerCamelCase__     (     )			-> Optional[int]:



					'''simple docstring'''




					_snake_case									=				get_version()
					_snake_case									=				F'''{current_version.major}.{current_version.minor + 1}.0.dev0'''
					_snake_case									=				current_version.base_version

					# Check with the user we got that right.
					_snake_case									=				input(F'''Which version are we developing now? [{dev_version}]'''						)
					if len(UpperCamelCase__						) == 0:
										_snake_case									=				dev_version

					print(F'''Updating version to {version}.'''						)
					global_version_update(UpperCamelCase__						)


#    print("Cleaning main README, don't forget to run `make fix-copies`.")
#    clean_main_ref_in_model_list()


if __name__ == "__main__":
					UpperCAmelCase_      							=     argparse.ArgumentParser()
					parser.add_argument("""--post_release""", action="""store_true""", help="""Whether this is pre or post release.""")
					parser.add_argument("""--patch""", action="""store_true""", help="""Whether or not this is a patch release.""")
					UpperCAmelCase_      							=     parser.parse_args()
					if not args.post_release:
										pre_release_work(patch=args.patch)
					elif args.patch:
										print("""Nothing to do after a patch :-)""")
					else:
										post_release_work()










import copy
import tempfile
import unittest

from huggingface_hub import HfFolder, delete_repo
from parameterized import parameterized
from requests.exceptions import HTTPError

from transformers import AutoConfig, GenerationConfig
from transformers.testing_utils import TOKEN, USER, is_staging_test
class 			UpperCamelCase_		(  unittest.TestCase					):




					@parameterized.expand([(None,), ('foo.json',)]  )
					def 	lowerCAmelCase			(		self				,						lowerCAmelCase_  )		->							str:
										_snake_case									=				GenerationConfig(
										    do_sample=lowerCAmelCase_				,						temperature=0.7				,						length_penalty=1.0				,						bad_words_ids=[[1, 2, 3], [4, 5]]				,						)
										with tempfile.TemporaryDirectory() as tmp_dir:
															config.save_pretrained(lowerCAmelCase_				,						config_name=lowerCAmelCase_  )
															_snake_case									=				GenerationConfig.from_pretrained(lowerCAmelCase_				,						config_name=lowerCAmelCase_  )

										# Checks parameters that were specified
										self.assertEqual(loaded_config.do_sample				,						lowerCAmelCase_  )
										self.assertEqual(loaded_config.temperature				,						0.7  )
										self.assertEqual(loaded_config.length_penalty				,						1.0  )
										self.assertEqual(loaded_config.bad_words_ids				,						[[1, 2, 3], [4, 5]]  )

										# Checks parameters that were not specified (defaults)
										self.assertEqual(loaded_config.top_k				,						50  )
										self.assertEqual(loaded_config.max_length				,						20  )
										self.assertEqual(loaded_config.max_time				,						lowerCAmelCase_  )




					def 	lowerCAmelCase			(		self  )		->							Optional[int]:
										_snake_case									=				AutoConfig.from_pretrained('gpt2'  )
										_snake_case									=				GenerationConfig.from_model_config(lowerCAmelCase_  )
										_snake_case									=				GenerationConfig()

										# The generation config has loaded a few non-default parameters from the model config
										self.assertNotEqual(lowerCAmelCase_				,						lowerCAmelCase_  )

										# One of those parameters is eos_token_id -- check if it matches
										self.assertNotEqual(generation_config_from_model.eos_token_id				,						default_generation_config.eos_token_id  )
										self.assertEqual(generation_config_from_model.eos_token_id				,						model_config.eos_token_id  )




					def 	lowerCAmelCase			(		self  )		->							Tuple:
										_snake_case									=				GenerationConfig()
										_snake_case									=				{
										    'max_new_tokens': 1024,
										    'foo': 'bar',
										}
										_snake_case									=				copy.deepcopy(lowerCAmelCase_  )
										_snake_case									=				generation_config.update(**lowerCAmelCase_  )

										# update_kwargs was not modified (no side effects)
										self.assertEqual(lowerCAmelCase_				,						lowerCAmelCase_  )

										# update_kwargs was used to update the config on valid attributes
										self.assertEqual(generation_config.max_new_tokens				,						1024  )

										# `.update()` returns a dictionary of unused kwargs
										self.assertEqual(lowerCAmelCase_				,						{'foo': 'bar'}  )




					def 	lowerCAmelCase			(		self  )		->							Optional[int]:
										_snake_case									=				GenerationConfig()
										_snake_case									=				'bar'

										with tempfile.TemporaryDirectory('test-generation-config'  ) as tmp_dir:
															generation_config.save_pretrained(lowerCAmelCase_  )

															_snake_case									=				GenerationConfig.from_pretrained(lowerCAmelCase_  )
										# update_kwargs was used to update the config on valid attributes
										self.assertEqual(new_config.foo				,						'bar'  )

										_snake_case									=				GenerationConfig.from_model_config(lowerCAmelCase_  )
										assert not hasattr(lowerCAmelCase_				,						'foo'  )  # no new kwargs should be initialized if from config






					def 	lowerCAmelCase			(		self  )		->							List[Any]:
										_snake_case									=				GenerationConfig()
										self.assertEqual(default_config.temperature				,						1.0  )
										self.assertEqual(default_config.do_sample				,						lowerCAmelCase_  )
										self.assertEqual(default_config.num_beams				,						1  )

										_snake_case									=				GenerationConfig(
										    do_sample=lowerCAmelCase_				,						temperature=0.7				,						length_penalty=1.0				,						bad_words_ids=[[1, 2, 3], [4, 5]]				,						)
										self.assertEqual(config.temperature				,						0.7  )
										self.assertEqual(config.do_sample				,						lowerCAmelCase_  )
										self.assertEqual(config.num_beams				,						1  )

										with tempfile.TemporaryDirectory() as tmp_dir:
															config.save_pretrained(lowerCAmelCase_  )
															_snake_case									=				GenerationConfig.from_pretrained(lowerCAmelCase_				,						temperature=1.0  )

										self.assertEqual(loaded_config.temperature				,						1.0  )
										self.assertEqual(loaded_config.do_sample				,						lowerCAmelCase_  )
										self.assertEqual(loaded_config.num_beams				,						1  )  # default value





@is_staging_test
class 			UpperCamelCase_		(  unittest.TestCase					):




					@classmethod
					def 	lowerCAmelCase			(		cls  )		->							List[str]:
										_snake_case									=				TOKEN
										HfFolder.save_token(lowerCAmelCase_  )




					@classmethod
					def 	lowerCAmelCase			(		cls  )		->							Union[str, Any]:
										try:
															delete_repo(token=cls._token				,						repo_id='test-generation-config'  )
										except HTTPError:
															pass

										try:
															delete_repo(token=cls._token				,						repo_id='valid_org/test-generation-config-org'  )
										except HTTPError:
															pass




					def 	lowerCAmelCase			(		self  )		->							int:
										_snake_case									=				GenerationConfig(
										    do_sample=lowerCAmelCase_				,						temperature=0.7				,						length_penalty=1.0				,						)
										config.push_to_hub('test-generation-config'				,						use_auth_token=self._token  )

										_snake_case									=				GenerationConfig.from_pretrained(F'''{USER}/test-generation-config'''  )
										for k, v in config.to_dict().items():
															if k != "transformers_version":
																				self.assertEqual(lowerCAmelCase_				,						getattr(lowerCAmelCase_				,						lowerCAmelCase_  )  )

        # Reset repo
										delete_repo(token=self._token				,						repo_id='test-generation-config'  )

										# Push to hub via save_pretrained
										with tempfile.TemporaryDirectory() as tmp_dir:
															config.save_pretrained(
															    lowerCAmelCase_				,						repo_id='test-generation-config'				,						push_to_hub=lowerCAmelCase_				,						use_auth_token=self._token  )

										_snake_case									=				GenerationConfig.from_pretrained(F'''{USER}/test-generation-config'''  )
										for k, v in config.to_dict().items():
															if k != "transformers_version":
																				self.assertEqual(lowerCAmelCase_				,						getattr(lowerCAmelCase_				,						lowerCAmelCase_  )  )






					def 	lowerCAmelCase			(		self  )		->							List[str]:
										_snake_case									=				GenerationConfig(
										    do_sample=lowerCAmelCase_				,						temperature=0.7				,						length_penalty=1.0				,						)
										config.push_to_hub('valid_org/test-generation-config-org'				,						use_auth_token=self._token  )

										_snake_case									=				GenerationConfig.from_pretrained('valid_org/test-generation-config-org'  )
										for k, v in config.to_dict().items():
															if k != "transformers_version":
																				self.assertEqual(lowerCAmelCase_				,						getattr(lowerCAmelCase_				,						lowerCAmelCase_  )  )

        # Reset repo
										delete_repo(token=self._token				,						repo_id='valid_org/test-generation-config-org'  )

										# Push to hub via save_pretrained
										with tempfile.TemporaryDirectory() as tmp_dir:
															config.save_pretrained(
															    lowerCAmelCase_				,						repo_id='valid_org/test-generation-config-org'				,						push_to_hub=lowerCAmelCase_				,						use_auth_token=self._token  )

										_snake_case									=				GenerationConfig.from_pretrained('valid_org/test-generation-config-org'  )
										for k, v in config.to_dict().items():
															if k != "transformers_version":
																				self.assertEqual(lowerCAmelCase_				,						getattr(lowerCAmelCase_				,						lowerCAmelCase_  )  )



