



'''simple docstring'''


# 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 torch

from ..models.auto import AutoModelForSequenceClassification, AutoTokenizer
from .base import PipelineTool




class      __UpperCamelCase       (_UpperCAmelCase			):
	__A			  =					'''facebook/bart-large-mnli'''
	__A			  =					(
	    '''This is a tool that classifies an English text using provided labels. It takes two inputs: `text`, which '''
	    '''should be the text to classify, and `labels`, which should be the list of labels to use for classification. '''
	    '''It returns the most likely label in the list of provided `labels` for the input text.'''
	)
	__A			  =					'''text_classifier'''
	__A			  =					AutoTokenizer
	__A			  =					AutoModelForSequenceClassification

	__A			  =					['''text''', ['''text''']]
	__A			  =					['''text''']







	def   _a   (    self						)    ->    Union[str, Any]:
							'''simple docstring'''






							super().setup()
							lowercase        =							self.model.config
							lowercase        =							-1
							for idx, label in config.idalabel.items():
													if label.lower().startswith("""entail"""						):
																			lowercase        =							int(_lowerCAmelCase						)
							if self.entailment_id == -1:
													raise ValueError("""Could not determine the entailment ID from the model config, please pass it at init."""						)







	def   _a   (    self ,				_lowerCAmelCase ,				_lowerCAmelCase						)    ->    List[str]:
							'''simple docstring'''






							lowercase        =							labels
							return self.pre_processor(
							    [text] * len(_lowerCAmelCase						) ,				[F"""This example is {label}""" for label in labels] ,				return_tensors="""pt""" ,				padding="""max_length""" ,				)







	def   _a   (    self ,				_lowerCAmelCase						)    ->    Tuple:
							'''simple docstring'''






							lowercase        =							outputs.logits
							lowercase        =							torch.argmax(logits[:, 2]						).item()
							return self._labels[label_id]







'''simple docstring'''


import json
import logging
import os
import socket

import git
import numpy as np
import torch


logging.basicConfig(
    format='''%(asctime)s - %(levelname)s - %(name)s - PID: %(process)d -  %(message)s''',
    datefmt='''%m/%d/%Y %H:%M:%S''',
    level=logging.INFO,
)
lowercase_      :				Tuple   					=    logging.getLogger(__name__)







def      SCREAMING_SNAKE_CASE				(				lowercase_    :      str						):
						lowercase        =							git.Repo(search_parent_directories=lowercase_						)
						lowercase        =							{
						    """repo_id""": str(lowercase_						),
						    """repo_sha""": str(repo.head.object.hexsha						),
						    """repo_branch""": str(repo.active_branch						),
						}

						with open(os.path.join(lowercase_ ,  """git_log.json"""						) ,  """w"""						) as f:
												json.dump(lowercase_ ,  lowercase_ ,  indent=4						)







def      SCREAMING_SNAKE_CASE				(				lowercase_    :      str						):
						if params.n_gpu <= 0:
												lowercase        =							0
												lowercase        =							-1
												lowercase        =							True
												lowercase        =							False
												return

						assert torch.cuda.is_available()

						logger.info("""Initializing GPUs"""						)
						if params.n_gpu > 1:
												assert params.local_rank != -1

												lowercase        =							int(os.environ["""WORLD_SIZE"""]						)
												lowercase        =							int(os.environ["""N_GPU_NODE"""]						)
												lowercase        =							int(os.environ["""RANK"""]						)

												# number of nodes / node ID
												lowercase        =							params.world_size // params.n_gpu_per_node
												lowercase        =							params.global_rank // params.n_gpu_per_node
												lowercase        =							True

												assert params.n_nodes == int(os.environ["""N_NODES"""]						)
												assert params.node_id == int(os.environ["""NODE_RANK"""]						)

						# local job (single GPU)
						else:
												assert params.local_rank == -1

												lowercase        =							1
												lowercase        =							0
												lowercase        =							0
												lowercase        =							0
												lowercase        =							1
												lowercase        =							1
												lowercase        =							False

						# sanity checks
						assert params.n_nodes >= 1
						assert 0 <= params.node_id < params.n_nodes
						assert 0 <= params.local_rank <= params.global_rank < params.world_size
						assert params.world_size == params.n_nodes * params.n_gpu_per_node

						# define whether this is the master process / if we are in multi-node distributed mode
						lowercase        =							params.node_id == 0 and params.local_rank == 0
						lowercase        =							params.n_nodes > 1

						# summary
						lowercase        =							F"""--- Global rank: {params.global_rank} - """
						logger.info(PREFIX + """Number of nodes: %i""" % params.n_nodes						)
						logger.info(PREFIX + """Node ID        : %i""" % params.node_id						)
						logger.info(PREFIX + """Local rank     : %i""" % params.local_rank						)
						logger.info(PREFIX + """World size     : %i""" % params.world_size						)
						logger.info(PREFIX + """GPUs per node  : %i""" % params.n_gpu_per_node						)
						logger.info(PREFIX + """Master         : %s""" % str(params.is_master						)						)
						logger.info(PREFIX + """Multi-node     : %s""" % str(params.multi_node						)						)
						logger.info(PREFIX + """Multi-GPU      : %s""" % str(params.multi_gpu						)						)
						logger.info(PREFIX + """Hostname       : %s""" % socket.gethostname()						)

						# set GPU device
						torch.cuda.set_device(params.local_rank						)

						# initialize multi-GPU
						if params.multi_gpu:
												logger.info("""Initializing PyTorch distributed"""						)
												torch.distributed.init_process_group(
												    init_method="""env://""" ,  backend="""nccl""" ,  )







def      SCREAMING_SNAKE_CASE				(				lowercase_    :      Optional[Any]						):
						np.random.seed(args.seed						)
						torch.manual_seed(args.seed						)
						if args.n_gpu > 0:
												torch.cuda.manual_seed_all(args.seed						)


