


'''simple docstring'''






import unittest

from transformers import is_torch_available
from transformers.testing_utils import require_sentencepiece, require_tokenizers, require_torch, slow


if is_torch_available():
 import torch

 from transformers import XLMRobertaModel




@require_sentencepiece
@require_tokenizers
@require_torch
class  a     (     unittest.TestCase       ):
 @slow
 def 							__UpperCAmelCase		( self						)					->							Tuple:
      _a			      =							XLMRobertaModel.from_pretrained('xlm-roberta-base'						)
      _a			      =							torch.tensor([[0, 5_81, 1_02_69, 83, 9_99_42, 1_36, 6_07_42, 23, 70, 8_05_83, 1_82_76, 2]]						)
      # The dog is cute and lives in the garden house

      _a			      =							torch.Size((1, 12, 7_68)						)  # batch_size, sequence_length, embedding_vector_dim
      _a			      =							torch.tensor(
          [[-0.0_1_0_1, 0.1_2_1_8, -0.0_8_0_3, 0.0_8_0_1, 0.1_3_2_7, 0.0_7_7_6, -0.1_2_1_5, 0.2_3_8_3, 0.3_3_3_8, 0.3_1_0_6, 0.0_3_0_0, 0.0_2_5_2]]						)
      #  xlmr = torch.hub.load('pytorch/fairseq', 'xlmr.base')
      #  xlmr.eval()
      #  expected_output_values_last_dim = xlmr.extract_features(input_ids[0])[:, :, -1]
      with torch.no_grad():
           _a			      =							model(__magic_name__						)['last_hidden_state'].detach()
      self.assertEqual(output.shape      ,				__magic_name__						)
      # compare the actual values for a slice of last dim
      self.assertTrue(torch.allclose(output[:, :, -1]      ,				__magic_name__      ,				atol=1e-3						)						)




 @slow
 def 							__UpperCAmelCase		( self						)					->							Any:
      _a			      =							XLMRobertaModel.from_pretrained('xlm-roberta-large'						)
      _a			      =							torch.tensor([[0, 5_81, 1_02_69, 83, 9_99_42, 1_36, 6_07_42, 23, 70, 8_05_83, 1_82_76, 2]]						)
      # The dog is cute and lives in the garden house

      _a			      =							torch.Size((1, 12, 10_24)						)  # batch_size, sequence_length, embedding_vector_dim
      _a			      =							torch.tensor(
          [[-0.0_6_9_9, -0.0_3_1_8, 0.0_7_0_5, -0.1_2_4_1, 0.0_9_9_9, -0.0_5_2_0, 0.1_0_0_4, -0.1_8_3_8, -0.4_7_0_4, 0.1_4_3_7, 0.0_8_2_1, 0.0_1_2_6]]						)
      #  xlmr = torch.hub.load('pytorch/fairseq', 'xlmr.large')
      #  xlmr.eval()
      #  expected_output_values_last_dim = xlmr.extract_features(input_ids[0])[:, :, -1]
      with torch.no_grad():
           _a			      =							model(__magic_name__						)['last_hidden_state'].detach()
      self.assertEqual(output.shape      ,				__magic_name__						)
      # compare the actual values for a slice of last dim
      self.assertTrue(torch.allclose(output[:, :, -1]      ,				__magic_name__      ,				atol=1e-3						)						)





'''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 typing import TYPE_CHECKING

import torch

from ..models.auto import AutoModelForVisualQuestionAnswering, AutoProcessor
from ..utils import requires_backends
from .base import PipelineTool


if TYPE_CHECKING:
 from PIL import Image




class  a     (     _SCREAMING_SNAKE_CASE       ):
 _lowerCAmelCase 					=	"""dandelin/vilt-b32-finetuned-vqa"""
 _lowerCAmelCase 					=	(
     """This is a tool that answers a question about an image. It takes an input named `image` which should be the """
     """image containing the information, as well as a `question` which should be the question in English. It """
     """returns a text that is the answer to the question."""
 )
 _lowerCAmelCase 					=	"""image_qa"""
 _lowerCAmelCase 					=	AutoProcessor
 _lowerCAmelCase 					=	AutoModelForVisualQuestionAnswering

 _lowerCAmelCase 					=	["""image""", """text"""]
 _lowerCAmelCase 					=	["""text"""]
 def __init__( self      ,				*__magic_name__      ,				**__magic_name__						)					->							Tuple:
      requires_backends(self      ,				['vision']						)
      super().__init__(*__magic_name__      ,				**__magic_name__						)
 def 							__UpperCAmelCase		( self      ,				__magic_name__      ,				__magic_name__						)					->							Tuple:
      return self.pre_processor(__magic_name__      ,				__magic_name__      ,				return_tensors='pt'						)
 def 							__UpperCAmelCase		( self      ,				__magic_name__						)					->							Any:
      with torch.no_grad():
           return self.model(**__magic_name__						).logits




 def 							__UpperCAmelCase		( self      ,				__magic_name__						)					->							Optional[int]:
      _a			      =							outputs.argmax(-1						).item()
      return self.model.config.idalabel[idx]

