


'''simple docstring'''




import os
import unittest

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


if is_torch_available():
				import torch

				from transformers import GPTaLMHeadModel





@require_torch
@require_sentencepiece
@require_tokenizers
class _snake_case (			unittest.TestCase ):
							"""simple docstring"""





							@slow
							@unittest.skip("Model is not available."						)
							def 		__SCREAMING_SNAKE_CASE( self						)  ->Union[str, Any]:



											"""simple docstring"""







											__lowerCamelCase		:   Any           =       "nvidia/megatron-gpt2-345m/"
											if "MYDIR" in os.environ:
															__lowerCamelCase		:   str           =       os.path.join(os.environ["MYDIR"]      ,					A_						)
											__lowerCamelCase		:   Optional[Any]           =       GPTaLMHeadModel.from_pretrained(A_						)
											model.to(A_						)
											model.half()

											__lowerCamelCase		:   Dict           =       torch.tensor(
											    [[101, 7110, 1005, 1056, 2023, 11333, 17413, 1029, 102]]      ,					device=A_      ,					dtype=torch.long      ,					)

											with torch.no_grad():
															__lowerCamelCase		:   List[Any]           =       model(A_						).logits

											__lowerCamelCase		:   int           =       torch.Size((1, 9, 50257)						)
											self.assertEqual(output.shape      ,					A_						)

											__lowerCamelCase		:   List[str]           =       torch.tensor(
											    [
											        4.9414,
											        -0.2920,
											        -1.2148,
											        -4.0273,
											        -0.5161,
											        -5.2109,
											        -1.2412,
											        -1.8301,
											        -1.7734,
											        -4.7148,
											        -0.2317,
											        -1.0811,
											        -2.1777,
											        0.4141,
											        -3.7969,
											        -4.0586,
											        -2.5332,
											        -3.3809,
											        4.3867,
											    ]      ,					device=A_      ,					dtype=torch.half      ,					)

											for i in range(19						):
															__lowerCamelCase    ,       __lowerCamelCase		:   List[Any]           =       8 * i // 17, 2792 * i  # along the diagonal
															__lowerCamelCase    ,       __lowerCamelCase		:   Dict           =       output[0, r, c], expected_diag[i]
															__lowerCamelCase		:   Dict           =       f'row={r} col={c} computed={computed} expected={expected}'
															self.assertAlmostEqual(A_      ,					A_      ,					delta=1E-4      ,					msg=A_						)









import unittest

import numpy as np

from transformers.testing_utils import require_torch, require_vision
from transformers.utils import is_torch_available, is_vision_available

from ...test_image_processing_common import ImageProcessingSavingTestMixin, prepare_image_inputs


if is_torch_available():
   import torch

if is_vision_available():
   from PIL import Image

   from transformers import EfficientNetImageProcessor






class        UpperCAmelCase__			(   unittest.TestCase						):



    """simple docstring"""






    def __init__(     self   ,  A_   ,  A_=13   ,  A_=3   ,  A_=18   ,  A_=30   ,  A_=400   ,  A_=True   ,  A_=None   ,  A_=True   ,  A_=[0.5, 0.5, 0.5]   ,  A_=[0.5, 0.5, 0.5]   ,  )    ->			int:
         __UpperCamelCase       				=size if size is not None else {'height': 18, 'width': 18}
         __UpperCamelCase       				=parent
         __UpperCamelCase       				=batch_size
         __UpperCamelCase       				=num_channels
         __UpperCamelCase       				=image_size
         __UpperCamelCase       				=min_resolution
         __UpperCamelCase       				=max_resolution
         __UpperCamelCase       				=do_resize
         __UpperCamelCase       				=size
         __UpperCamelCase       				=do_normalize
         __UpperCamelCase       				=image_mean
         __UpperCamelCase       				=image_std






    def _a   (     self       )    ->			Optional[Any]:
         return {
             "image_mean": self.image_mean,
             "image_std": self.image_std,
             "do_normalize": self.do_normalize,
             "do_resize": self.do_resize,
             "size": self.size,
         }






@require_torch
@require_vision
class        UpperCAmelCase__			(   A_  ,    unittest.TestCase						):



    """simple docstring"""






    UpperCAmelCase__		:						List[Any]    =    EfficientNetImageProcessor if is_vision_available() else None






    def _a   (     self       )    ->			List[str]:
         __UpperCamelCase       				=EfficientNetImageProcessorTester(self       )






    @property
    def _a   (     self       )    ->			Optional[Any]:
         return self.image_processor_tester.prepare_image_processor_dict()






    def _a   (     self       )    ->			str:
         __UpperCamelCase       				=self.image_processing_class(**self.image_processor_dict       )
         self.assertTrue(hasattr(A_   ,  'image_mean'       )       )
         self.assertTrue(hasattr(A_   ,  'image_std'       )       )
         self.assertTrue(hasattr(A_   ,  'do_normalize'       )       )
         self.assertTrue(hasattr(A_   ,  'do_resize'       )       )
         self.assertTrue(hasattr(A_   ,  'size'       )       )






    def _a   (     self       )    ->			Dict:
         __UpperCamelCase       				=self.image_processing_class.from_dict(self.image_processor_dict       )
         self.assertEqual(image_processor.size   ,  {'height': 18, 'width': 18}       )

         __UpperCamelCase       				=self.image_processing_class.from_dict(self.image_processor_dict   ,  size=42       )
         self.assertEqual(image_processor.size   ,  {'height': 42, 'width': 42}       )






    def _a   (     self       )    ->			List[str]:
         # Initialize image_processing
         __UpperCamelCase       				=self.image_processing_class(**self.image_processor_dict       )
         # create random PIL images
         __UpperCamelCase       				=prepare_image_inputs(self.image_processor_tester   ,  equal_resolution=A_       )
         for image in image_inputs:
              self.assertIsInstance(A_   ,  Image.Image       )

         # Test not batched input
         __UpperCamelCase       				=image_processing(image_inputs[0]   ,  return_tensors='pt'       ).pixel_values
         self.assertEqual(
             encoded_images.shape   ,  (
                 1,
                 self.image_processor_tester.num_channels,
                 self.image_processor_tester.size['height'],
                 self.image_processor_tester.size['width'],
             )   ,  )

         # Test batched
         __UpperCamelCase       				=image_processing(A_   ,  return_tensors='pt'       ).pixel_values
         self.assertEqual(
             encoded_images.shape   ,  (
                 self.image_processor_tester.batch_size,
                 self.image_processor_tester.num_channels,
                 self.image_processor_tester.size['height'],
                 self.image_processor_tester.size['width'],
             )   ,  )






    def _a   (     self       )    ->			List[str]:
         # Initialize image_processing
         __UpperCamelCase       				=self.image_processing_class(**self.image_processor_dict       )
         # create random numpy tensors
         __UpperCamelCase       				=prepare_image_inputs(self.image_processor_tester   ,  equal_resolution=A_   ,  numpify=A_       )
         for image in image_inputs:
              self.assertIsInstance(A_   ,  np.ndarray       )

         # Test not batched input
         __UpperCamelCase       				=image_processing(image_inputs[0]   ,  return_tensors='pt'       ).pixel_values
         self.assertEqual(
             encoded_images.shape   ,  (
                 1,
                 self.image_processor_tester.num_channels,
                 self.image_processor_tester.size['height'],
                 self.image_processor_tester.size['width'],
             )   ,  )

         # Test batched
         __UpperCamelCase       				=image_processing(A_   ,  return_tensors='pt'       ).pixel_values
         self.assertEqual(
             encoded_images.shape   ,  (
                 self.image_processor_tester.batch_size,
                 self.image_processor_tester.num_channels,
                 self.image_processor_tester.size['height'],
                 self.image_processor_tester.size['width'],
             )   ,  )






    def _a   (     self       )    ->			Dict:
         # Initialize image_processing
         __UpperCamelCase       				=self.image_processing_class(**self.image_processor_dict       )
         # create random PyTorch tensors
         __UpperCamelCase       				=prepare_image_inputs(self.image_processor_tester   ,  equal_resolution=A_   ,  torchify=A_       )
         for image in image_inputs:
              self.assertIsInstance(A_   ,  torch.Tensor       )

         # Test not batched input
         __UpperCamelCase       				=image_processing(image_inputs[0]   ,  return_tensors='pt'       ).pixel_values
         self.assertEqual(
             encoded_images.shape   ,  (
                 1,
                 self.image_processor_tester.num_channels,
                 self.image_processor_tester.size['height'],
                 self.image_processor_tester.size['width'],
             )   ,  )

         # Test batched
         __UpperCamelCase       				=image_processing(A_   ,  return_tensors='pt'       ).pixel_values
         self.assertEqual(
             encoded_images.shape   ,  (
                 self.image_processor_tester.batch_size,
                 self.image_processor_tester.num_channels,
                 self.image_processor_tester.size['height'],
                 self.image_processor_tester.size['width'],
             )   ,  )





