


'''simple docstring'''

import argparse
import os


lowerCamelCase							:       Tuple  						=     {
    'pytorch': {
        '1.13': {
            'torch': '1.13.1',
            'torchvision': '0.14.1',
            'torchaudio': '0.13.1',
            'python': 3.9,
            'cuda': 'cu116',
            'install': (
                'python3 -m pip install --no-cache-dir -U torch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1'
                ' --extra-index-url https://download.pytorch.org/whl/cu116'
            ),
            'base_image': 'nvidia/cuda:11.6.2-cudnn8-devel-ubuntu20.04',
        },
        '1.12': {
            'torch': '1.12.1',
            'torchvision': '0.13.1',
            'torchaudio': '0.12.1',
            'python': 3.9,
            'cuda': 'cu113',
            'install': (
                'python3 -m pip install --no-cache-dir -U torch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1'
                ' --extra-index-url https://download.pytorch.org/whl/cu113'
            ),
            'base_image': 'nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04',
        },
        '1.11': {
            'torch': '1.11.0',
            'torchvision': '0.12.0',
            'torchaudio': '0.11.0',
            'python': 3.9,
            'cuda': 'cu113',
            'install': (
                'python3 -m pip install --no-cache-dir -U torch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0'
                ' --extra-index-url https://download.pytorch.org/whl/cu113'
            ),
            'base_image': 'nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04',
        },
        '1.10': {
            'torch': '1.10.2',
            'torchvision': '0.11.3',
            'torchaudio': '0.10.2',
            'python': 3.9,
            'cuda': 'cu113',
            'install': (
                'python3 -m pip install --no-cache-dir -U torch==1.10.2 torchvision==0.11.3 torchaudio==0.10.2'
                ' --extra-index-url https://download.pytorch.org/whl/cu113'
            ),
            'base_image': 'nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04',
        },
        # torchaudio < 0.10 has no CUDA-enabled binary distributions
        '1.9': {
            'torch': '1.9.1',
            'torchvision': '0.10.1',
            'torchaudio': '0.9.1',
            'python': 3.9,
            'cuda': 'cu111',
            'install': (
                'python3 -m pip install --no-cache-dir -U torch==1.9.1 torchvision==0.10.1 torchaudio==0.9.1'
                ' --extra-index-url https://download.pytorch.org/whl/cu111'
            ),
            'base_image': 'nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04',
        },
    },
    'tensorflow': {
        '2.11': {
            'tensorflow': '2.11.1',
            'install': 'python3 -m pip install --no-cache-dir -U tensorflow==2.11.1',
            'base_image': 'nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04',
        },
        '2.10': {
            'tensorflow': '2.10.1',
            'install': 'python3 -m pip install --no-cache-dir -U tensorflow==2.10.1',
            'base_image': 'nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04',
        },
        '2.9': {
            'tensorflow': '2.9.3',
            'install': 'python3 -m pip install --no-cache-dir -U tensorflow==2.9.3',
            'base_image': 'nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04',
        },
        '2.8': {
            'tensorflow': '2.8.2',
            'install': 'python3 -m pip install --no-cache-dir -U tensorflow==2.8.2',
            'base_image': 'nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04',
        },
        '2.7': {
            'tensorflow': '2.7.3',
            'install': 'python3 -m pip install --no-cache-dir -U tensorflow==2.7.3',
            'base_image': 'nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04',
        },
        '2.6': {
            'tensorflow': '2.6.5',
            'install': 'python3 -m pip install --no-cache-dir -U tensorflow==2.6.5',
            'base_image': 'nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04',
        },
        '2.5': {
            'tensorflow': '2.5.3',
            'install': 'python3 -m pip install --no-cache-dir -U tensorflow==2.5.3',
            'base_image': 'nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04',
        },
    },
}


if __name__ == "__main__":
 lowerCamelCase							:       List[str]  						=     argparse.ArgumentParser('Choose the framework and version to install')
 parser.add_argument(
     '--framework', help='The framework to install. Should be `torch` or `tensorflow`', type=str, required=True
 )
 parser.add_argument('--version', help='The version of the framework to install.', type=str, required=True)
 lowerCamelCase							:       Optional[int]  						=     parser.parse_args()

 lowerCamelCase							:       Any  						=     past_versions_testing[args.framework][args.version]

 os.system(f"""echo \"export INSTALL_CMD=\'{info['install']}\'\" >> ~/.profile""")
 print(f"""echo \"export INSTALL_CMD=\'{info['install']}\'\" >> ~/.profile""")

 lowerCamelCase							:       List[Any]  						=     ''
 if args.framework == "pytorch":
  lowerCamelCase							:       Union[str, Any]  						=     info['cuda']
 os.system(f"""echo \"export CUDA='{cuda}'\" >> ~/.profile""")
 print(f"""echo \"export CUDA='{cuda}'\" >> ~/.profile""")








'''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.
from ..models.auto import AutoModelForSeqaSeqLM, AutoTokenizer
from .base import PipelineTool


lowerCamelCase							:       Any  						=     'Here is a text containing a lot of information: \'\'\'{text}\'\'\'.\n\nCan you answer this question about the text: \'{question}\''
class   __lowerCAmelCase						(lowercase_					):





      '''simple docstring'''






      lowerCAmelCase__ :					Dict   =							"""google/flan-t5-base"""
      lowerCAmelCase__ :					List[Any]   =							(
          """This is a tool that answers questions related to a text. It takes two arguments named `text`, which is the """
          """text where to find the answer, and `question`, which is the question, and returns the answer to the question."""
      )
      lowerCAmelCase__ :					Any   =							"""text_qa"""
      lowerCAmelCase__ :					Union[str, Any]   =							AutoTokenizer
      lowerCAmelCase__ :					Any   =							AutoModelForSeqaSeqLM

      lowerCAmelCase__ :					Any   =							["""text""", """text"""]
      lowerCAmelCase__ :					Optional[int]   =							["""text"""]



      def 				UpperCamelCase__		(self   :   Union[str, Any]       ,						UpperCamelCase   :   str       ,						UpperCamelCase   :   str			):

             '''simple docstring'''







             lowercase__				      =			QA_PROMPT.format(text=UpperCamelCase       ,						question=UpperCamelCase			)
             return self.pre_processor(UpperCamelCase       ,						return_tensors='''pt'''			)



      def 				UpperCamelCase__		(self   :   Dict       ,						UpperCamelCase   :   List[Any]			):

             '''simple docstring'''







             lowercase__				      =			self.model.generate(**UpperCamelCase			)

             lowercase__      ,lowercase__				      =			inputs['''input_ids'''].shape
             lowercase__				      =			output_ids.shape[0]

             return output_ids.reshape(UpperCamelCase       ,						out_b // in_b       ,						*output_ids.shape[1:]			)[0][0]






      def 				UpperCamelCase__		(self   :   Tuple       ,						UpperCamelCase   :   Optional[Any]			):

             '''simple docstring'''







             return self.pre_processor.decode(UpperCamelCase       ,						skip_special_tokens=UpperCamelCase       ,						clean_up_tokenization_spaces=UpperCamelCase			)




