
# Copyright 2023 The HuggingFace Inc. team. All rights reserved.
#
# 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.
import re

from ..utils import cached_file


# docstyle-ignore
__lowerCAmelCase     :	int       			=  '\nHuman: <<task>>\n\nAssistant: '


__lowerCAmelCase     :	str       			=  'huggingface-tools/default-prompts'
__lowerCAmelCase     :	Tuple       			=  {'chat': 'chat_prompt_template.txt', 'run': 'run_prompt_template.txt'}

def a__       (			A_,	A_,	A_="run"   ):






  '''simple docstring'''





  if prompt_or_repo_id is None:
    __magic_name__								=			DEFAULT_PROMPTS_REPO

  # prompt is considered a repo ID when it does not contain any kind of space
  if re.search("""\\s""",	A_   ) is not None:
    return prompt_or_repo_id

  __magic_name__								=			cached_file(
      A_,	PROMPT_FILES[mode],	repo_type="""dataset""",	user_agent={"""agent""": agent_name}   )
  with open(A_,	"""r""",	encoding="""utf-8"""   ) as f:
    return f.read()



from ...configuration_utils import PretrainedConfig
from ...utils import logging


__lowerCAmelCase     :	Dict       			=  logging.get_logger(__name__)

__lowerCAmelCase     :	int       			=  {
    'microsoft/trocr-base-handwritten': (
        'https://huggingface.co/microsoft/trocr-base-handwritten/resolve/main/config.json'
    ),
    # See all TrOCR models at https://huggingface.co/models?filter=trocr
}




class 			UpperCAmelCase_     (					_A	):


  '''simple docstring'''







  a__       			=						"""trocr"""
  a__       			=						["""past_key_values"""]
  a__       			=						{
      """num_attention_heads""": """decoder_attention_heads""",
      """hidden_size""": """d_model""",
      """num_hidden_layers""": """decoder_layers""",
  }
  def __init__(					self     :					Dict     ,				UpperCamelCase__     :					Optional[Any]=5_0265     ,				UpperCamelCase__     :					int=1024     ,				UpperCamelCase__     :					int=12     ,				UpperCamelCase__     :					List[Any]=16     ,				UpperCamelCase__     :					Union[str, Any]=4096     ,				UpperCamelCase__     :					List[Any]="gelu"     ,				UpperCamelCase__     :					Optional[int]=512     ,				UpperCamelCase__     :					Optional[Any]=0.1     ,				UpperCamelCase__     :					int=0.0     ,				UpperCamelCase__     :					str=0.0     ,				UpperCamelCase__     :					Optional[int]=2     ,				UpperCamelCase__     :					List[str]=0.02     ,				UpperCamelCase__     :					Dict=0.0     ,				UpperCamelCase__     :					List[str]=True     ,				UpperCamelCase__     :					str=False     ,				UpperCamelCase__     :					List[str]=True     ,				UpperCamelCase__     :					Tuple=True     ,				UpperCamelCase__     :					str=1     ,				UpperCamelCase__     :					Union[str, Any]=0     ,				UpperCamelCase__     :					List[str]=2     ,				**UpperCamelCase__     :					Tuple     ,				)     ->      Dict:







    """simple docstring"""






    __magic_name__								=			vocab_size
    __magic_name__								=			d_model
    __magic_name__								=			decoder_layers
    __magic_name__								=			decoder_attention_heads
    __magic_name__								=			decoder_ffn_dim
    __magic_name__								=			activation_function
    __magic_name__								=			max_position_embeddings
    __magic_name__								=			dropout
    __magic_name__								=			attention_dropout
    __magic_name__								=			activation_dropout
    __magic_name__								=			init_std
    __magic_name__								=			decoder_layerdrop
    __magic_name__								=			use_cache
    __magic_name__								=			scale_embedding
    __magic_name__								=			use_learned_position_embeddings
    __magic_name__								=			layernorm_embedding

    super().__init__(
        pad_token_id=UpperCamelCase__     ,				bos_token_id=UpperCamelCase__     ,				eos_token_id=UpperCamelCase__     ,				decoder_start_token_id=UpperCamelCase__     ,				**UpperCamelCase__     ,				)

