
from typing import Optional, Tuple

import jax
import jax.numpy as jnp
from flax import linen as nn
from flax.core.frozen_dict import FrozenDict
from transformers import CLIPConfig, FlaxPreTrainedModel
from transformers.models.clip.modeling_flax_clip import FlaxCLIPVisionModule






def 							_lowercase							(     UpperCamelCase_	,	UpperCamelCase_	,	UpperCamelCase_=1e-12		)		->		Optional[int]:
					'''simple docstring'''


					SCREAMING_SNAKE_CASE__													= jnp.divide(emb_a.T	,	jnp.clip(jnp.linalg.norm(UpperCamelCase_	,	axis=1		)	,	a_min=UpperCamelCase_		)		).T
					SCREAMING_SNAKE_CASE__													= jnp.divide(emb_a.T	,	jnp.clip(jnp.linalg.norm(UpperCamelCase_	,	axis=1		)	,	a_min=UpperCamelCase_		)		).T
					return jnp.matmul(UpperCamelCase_	,	norm_emb_a.T		)




class 						lowercase__  (						nn.Module  ):
	A__    :   CLIPConfig
	A__    :   jnp.dtype				       =jnp.floataa
	def      A_				(			self							:   int							):
						SCREAMING_SNAKE_CASE__													= FlaxCLIPVisionModule(self.config.vision_config							)
						SCREAMING_SNAKE_CASE__													= nn.Dense(self.config.projection_dim       ,   use_bias=UpperCAmelCase_       ,   dtype=self.dtype							)

						SCREAMING_SNAKE_CASE__													= self.param('concept_embeds'       ,   jax.nn.initializers.ones       ,   (17, self.config.projection_dim)							)
						SCREAMING_SNAKE_CASE__													= self.param(
						    'special_care_embeds'       ,   jax.nn.initializers.ones       ,   (3, self.config.projection_dim)							)

						SCREAMING_SNAKE_CASE__													= self.param('concept_embeds_weights'       ,   jax.nn.initializers.ones       ,   (17,)							)
						SCREAMING_SNAKE_CASE__													= self.param('special_care_embeds_weights'       ,   jax.nn.initializers.ones       ,   (3,)							)





	def __call__(			self							:   Any       ,   UpperCAmelCase_							:   int							):
						SCREAMING_SNAKE_CASE__													= self.vision_model(UpperCAmelCase_							)[1]
						SCREAMING_SNAKE_CASE__													= self.visual_projection(UpperCAmelCase_							)

						SCREAMING_SNAKE_CASE__													= jax_cosine_distance(UpperCAmelCase_       ,   self.special_care_embeds							)
						SCREAMING_SNAKE_CASE__													= jax_cosine_distance(UpperCAmelCase_       ,   self.concept_embeds							)

						# increase this value to create a stronger `nfsw` filter
						# at the cost of increasing the possibility of filtering benign image inputs
						SCREAMING_SNAKE_CASE__													= 0.0

						SCREAMING_SNAKE_CASE__													= special_cos_dist - self.special_care_embeds_weights[None, :] + adjustment
						SCREAMING_SNAKE_CASE__													= jnp.round(UpperCAmelCase_       ,   3							)
						SCREAMING_SNAKE_CASE__													= jnp.any(special_scores > 0       ,   axis=1       ,   keepdims=UpperCAmelCase_							)
						# Use a lower threshold if an image has any special care concept
						SCREAMING_SNAKE_CASE__													= is_special_care * 0.01

						SCREAMING_SNAKE_CASE__													= cos_dist - self.concept_embeds_weights[None, :] + special_adjustment
						SCREAMING_SNAKE_CASE__													= jnp.round(UpperCAmelCase_       ,   3							)
						SCREAMING_SNAKE_CASE__													= jnp.any(concept_scores > 0       ,   axis=1							)

						return has_nsfw_concepts





class 						lowercase__  (						_UpperCAmelCase  ):
	A__    :   List[str]				       =CLIPConfig
	A__    :   str				       ="""clip_input"""
	A__    :   Tuple				       =FlaxStableDiffusionSafetyCheckerModule
	def __init__(			self							:   Union[str, Any]       ,   UpperCAmelCase_							:   CLIPConfig       ,   UpperCAmelCase_							:   Optional[Tuple] = None       ,   UpperCAmelCase_							:   int = 0       ,   UpperCAmelCase_							:   jnp.dtype = jnp.floataa       ,   UpperCAmelCase_							:   bool = True       ,   **UpperCAmelCase_							:   int       ,   ):
						if input_shape is None:
											SCREAMING_SNAKE_CASE__													= (1, 224, 224, 3)
						SCREAMING_SNAKE_CASE__													= self.module_class(config=UpperCAmelCase_       ,   dtype=UpperCAmelCase_       ,   **UpperCAmelCase_							)
						super().__init__(UpperCAmelCase_       ,   UpperCAmelCase_       ,   input_shape=UpperCAmelCase_       ,   seed=UpperCAmelCase_       ,   dtype=UpperCAmelCase_       ,   _do_init=_do_init							)
	def      A_				(			self							:   Tuple       ,   UpperCAmelCase_							:   jax.random.KeyArray       ,   UpperCAmelCase_							:   Tuple       ,   UpperCAmelCase_							:   FrozenDict = None							):
						# init input tensor
						SCREAMING_SNAKE_CASE__													= jax.random.normal(UpperCAmelCase_       ,   UpperCAmelCase_							)

						SCREAMING_SNAKE_CASE__						, SCREAMING_SNAKE_CASE__													= jax.random.split(UpperCAmelCase_							)
						SCREAMING_SNAKE_CASE__													= {'params': params_rng, 'dropout': dropout_rng}

						SCREAMING_SNAKE_CASE__													= self.module.init(UpperCAmelCase_       ,   UpperCAmelCase_							)['params']

						return random_params





	def __call__(			self							:   int       ,   UpperCAmelCase_							:   int       ,   UpperCAmelCase_							:   dict = None       ,   ):
						SCREAMING_SNAKE_CASE__													= jnp.transpose(UpperCAmelCase_       ,   (0, 2, 3, 1)							)

						return self.module.apply(
						    {'params': params or self.params}       ,   jnp.array(UpperCAmelCase_       ,   dtype=jnp.floataa							)       ,   rngs={}       ,   )







import unittest

from transformers import DebertaVaTokenizer, DebertaVaTokenizerFast
from transformers.testing_utils import get_tests_dir, require_sentencepiece, require_tokenizers, slow

from ...test_tokenization_common import TokenizerTesterMixin


__snake_case								=  get_tests_dir("""fixtures/spiece.model""")




@require_sentencepiece
@require_tokenizers
class 						lowercase__  (						_UpperCAmelCase     ,       unittest.TestCase  ):
	A__    :   str				       =DebertaVaTokenizer
	A__    :   List[str]				       =DebertaVaTokenizerFast
	A__    :   Any				       =True
	A__    :   str				       =True
	def      A_				(			self							:   Tuple							):
						super().setUp()

						# We have a SentencePiece fixture for testing
						SCREAMING_SNAKE_CASE__													= DebertaVaTokenizer(UpperCAmelCase_       ,   unk_token='<unk>'							)
						tokenizer.save_pretrained(self.tmpdirname							)
	def      A_				(			self							:   Union[str, Any]       ,   UpperCAmelCase_							:   Optional[Any]							):
						SCREAMING_SNAKE_CASE__													= 'this is a test'
						SCREAMING_SNAKE_CASE__													= 'this is a test'
						return input_text, output_text
	def      A_				(			self							:   Optional[Any]							):
						SCREAMING_SNAKE_CASE__													= '<pad>'
						SCREAMING_SNAKE_CASE__													= 0

						self.assertEqual(self.get_tokenizer()._convert_token_to_id(UpperCAmelCase_							)       ,   UpperCAmelCase_							)
						self.assertEqual(self.get_tokenizer()._convert_id_to_token(UpperCAmelCase_							)       ,   UpperCAmelCase_							)
	def      A_				(			self							:   Union[str, Any]							):
						SCREAMING_SNAKE_CASE__													= list(self.get_tokenizer().get_vocab().keys()							)
						self.assertEqual(vocab_keys[0]       ,   '<pad>'							)
						self.assertEqual(vocab_keys[1]       ,   '<unk>'							)
						self.assertEqual(vocab_keys[-1]       ,   '[PAD]'							)
						self.assertEqual(len(UpperCAmelCase_							)       ,   30001							)
	def      A_				(			self							:   str							):
						self.assertEqual(self.get_tokenizer().vocab_size       ,   30000							)
	def      A_				(			self							:   Optional[Any]							):
						# fmt: off
						SCREAMING_SNAKE_CASE__													= ' \tHeLLo!how  \n Are yoU?  '
						SCREAMING_SNAKE_CASE__													= ['▁hello', '!', 'how', '▁are', '▁you', '?']
						# fmt: on

						SCREAMING_SNAKE_CASE__													= DebertaVaTokenizer(UpperCAmelCase_       ,   do_lower_case=UpperCAmelCase_							)
						SCREAMING_SNAKE_CASE__													= tokenizer.convert_ids_to_tokens(tokenizer.encode(UpperCAmelCase_       ,   add_special_tokens=UpperCAmelCase_							)							)

						self.assertListEqual(UpperCAmelCase_       ,   UpperCAmelCase_							)

						SCREAMING_SNAKE_CASE__													= DebertaVaTokenizerFast(UpperCAmelCase_       ,   do_lower_case=UpperCAmelCase_							)
						SCREAMING_SNAKE_CASE__													= rust_tokenizer.convert_ids_to_tokens(rust_tokenizer.encode(UpperCAmelCase_       ,   add_special_tokens=UpperCAmelCase_							)							)

						self.assertListEqual(UpperCAmelCase_       ,   UpperCAmelCase_							)
	@unittest.skip('There is an inconsistency between slow and fast tokenizer due to a bug in the fast one.'							)
	def      A_				(			self							:   Any							):
						pass
	@unittest.skip('There is an inconsistency between slow and fast tokenizer due to a bug in the fast one.'							)
	def      A_				(			self							:   Tuple							):
						pass
	def      A_				(			self							:   List[str]							):
						# fmt: off
						SCREAMING_SNAKE_CASE__													= 'I was born in 92000, and this is falsé.'
						SCREAMING_SNAKE_CASE__													= ['▁', '<unk>', '▁was', '▁born', '▁in', '▁9', '2000', '▁', ',', '▁and', '▁this', '▁is', '▁fal', 's', '<unk>', '▁', '.', ]
						# fmt: on

						SCREAMING_SNAKE_CASE__													= DebertaVaTokenizer(UpperCAmelCase_       ,   split_by_punct=UpperCAmelCase_							)
						SCREAMING_SNAKE_CASE__													= tokenizer.convert_ids_to_tokens(tokenizer.encode(UpperCAmelCase_       ,   add_special_tokens=UpperCAmelCase_							)							)

						self.assertListEqual(UpperCAmelCase_       ,   UpperCAmelCase_							)

						SCREAMING_SNAKE_CASE__													= DebertaVaTokenizerFast(UpperCAmelCase_       ,   split_by_punct=UpperCAmelCase_							)
						SCREAMING_SNAKE_CASE__													= rust_tokenizer.convert_ids_to_tokens(rust_tokenizer.encode(UpperCAmelCase_       ,   add_special_tokens=UpperCAmelCase_							)							)

						self.assertListEqual(UpperCAmelCase_       ,   UpperCAmelCase_							)
	def      A_				(			self							:   List[str]							):
						# fmt: off
						SCREAMING_SNAKE_CASE__													= 'I was born in 92000, and this is falsé.'
						SCREAMING_SNAKE_CASE__													= ['▁i', '▁was', '▁born', '▁in', '▁9', '2000', '▁', ',', '▁and', '▁this', '▁is', '▁fal', 's', '<unk>', '▁', '.', ]
						# fmt: on

						SCREAMING_SNAKE_CASE__													= DebertaVaTokenizer(UpperCAmelCase_       ,   do_lower_case=UpperCAmelCase_       ,   split_by_punct=UpperCAmelCase_							)
						SCREAMING_SNAKE_CASE__													= tokenizer.convert_ids_to_tokens(tokenizer.encode(UpperCAmelCase_       ,   add_special_tokens=UpperCAmelCase_							)							)
						self.assertListEqual(UpperCAmelCase_       ,   UpperCAmelCase_							)

						SCREAMING_SNAKE_CASE__													= DebertaVaTokenizerFast(UpperCAmelCase_       ,   do_lower_case=UpperCAmelCase_       ,   split_by_punct=UpperCAmelCase_							)
						SCREAMING_SNAKE_CASE__													= rust_tokenizer.convert_ids_to_tokens(rust_tokenizer.encode(UpperCAmelCase_       ,   add_special_tokens=UpperCAmelCase_							)							)
						self.assertListEqual(UpperCAmelCase_       ,   UpperCAmelCase_							)
	def      A_				(			self							:   int							):
						# fmt: off
						SCREAMING_SNAKE_CASE__													= 'I was born in 92000, and this is falsé.'
						SCREAMING_SNAKE_CASE__													= ['▁i', '▁was', '▁born', '▁in', '▁9', '2000', ',', '▁and', '▁this', '▁is', '▁fal', 's', '<unk>', '.', ]
						# fmt: on

						SCREAMING_SNAKE_CASE__													= DebertaVaTokenizer(UpperCAmelCase_       ,   do_lower_case=UpperCAmelCase_       ,   split_by_punct=UpperCAmelCase_							)
						SCREAMING_SNAKE_CASE__													= tokenizer.convert_ids_to_tokens(tokenizer.encode(UpperCAmelCase_       ,   add_special_tokens=UpperCAmelCase_							)							)

						self.assertListEqual(UpperCAmelCase_       ,   UpperCAmelCase_							)

						SCREAMING_SNAKE_CASE__													= DebertaVaTokenizerFast(UpperCAmelCase_       ,   do_lower_case=UpperCAmelCase_       ,   split_by_punct=UpperCAmelCase_							)
						SCREAMING_SNAKE_CASE__													= rust_tokenizer.convert_ids_to_tokens(rust_tokenizer.encode(UpperCAmelCase_       ,   add_special_tokens=UpperCAmelCase_							)							)

						self.assertListEqual(UpperCAmelCase_       ,   UpperCAmelCase_							)
	def      A_				(			self							:   Tuple							):
						# fmt: off
						SCREAMING_SNAKE_CASE__													= 'I was born in 92000, and this is falsé.'
						SCREAMING_SNAKE_CASE__													= ['▁', '<unk>', '▁was', '▁born', '▁in', '▁9', '2000', '▁', ',', '▁and', '▁this', '▁is', '▁fal', 's', '<unk>', '▁', '.', ]
						# fmt: on

						SCREAMING_SNAKE_CASE__													= DebertaVaTokenizer(UpperCAmelCase_       ,   do_lower_case=UpperCAmelCase_       ,   split_by_punct=UpperCAmelCase_							)
						SCREAMING_SNAKE_CASE__													= tokenizer.convert_ids_to_tokens(tokenizer.encode(UpperCAmelCase_       ,   add_special_tokens=UpperCAmelCase_							)							)

						self.assertListEqual(UpperCAmelCase_       ,   UpperCAmelCase_							)

						SCREAMING_SNAKE_CASE__													= DebertaVaTokenizerFast(UpperCAmelCase_       ,   do_lower_case=UpperCAmelCase_       ,   split_by_punct=UpperCAmelCase_							)
						SCREAMING_SNAKE_CASE__													= rust_tokenizer.convert_ids_to_tokens(rust_tokenizer.encode(UpperCAmelCase_       ,   add_special_tokens=UpperCAmelCase_							)							)

						self.assertListEqual(UpperCAmelCase_       ,   UpperCAmelCase_							)
	def      A_				(			self							:   Any							):
						# fmt: off
						SCREAMING_SNAKE_CASE__													= ' \tHeLLo!how  \n Are yoU?  '
						SCREAMING_SNAKE_CASE__													= ['▁', '<unk>', 'e', '<unk>', 'o', '!', 'how', '▁', '<unk>', 're', '▁yo', '<unk>', '?']
						# fmt: on

						SCREAMING_SNAKE_CASE__													= DebertaVaTokenizer(UpperCAmelCase_       ,   do_lower_case=UpperCAmelCase_       ,   split_by_punct=UpperCAmelCase_							)
						SCREAMING_SNAKE_CASE__													= tokenizer.convert_ids_to_tokens(tokenizer.encode(UpperCAmelCase_       ,   add_special_tokens=UpperCAmelCase_							)							)

						self.assertListEqual(UpperCAmelCase_       ,   UpperCAmelCase_							)

						SCREAMING_SNAKE_CASE__													= DebertaVaTokenizerFast(UpperCAmelCase_       ,   do_lower_case=UpperCAmelCase_       ,   split_by_punct=UpperCAmelCase_							)
						SCREAMING_SNAKE_CASE__													= rust_tokenizer.convert_ids_to_tokens(rust_tokenizer.encode(UpperCAmelCase_       ,   add_special_tokens=UpperCAmelCase_							)							)

						self.assertListEqual(UpperCAmelCase_       ,   UpperCAmelCase_							)
	def      A_				(			self							:   int							):
						SCREAMING_SNAKE_CASE__													= self.get_tokenizer()
						SCREAMING_SNAKE_CASE__													= self.get_rust_tokenizer()

						SCREAMING_SNAKE_CASE__													= 'I was born in 92000, and this is falsé.'

						SCREAMING_SNAKE_CASE__													= tokenizer.convert_ids_to_tokens(tokenizer.encode(UpperCAmelCase_       ,   add_special_tokens=UpperCAmelCase_							)							)
						SCREAMING_SNAKE_CASE__													= rust_tokenizer.convert_ids_to_tokens(rust_tokenizer.encode(UpperCAmelCase_       ,   add_special_tokens=UpperCAmelCase_							)							)
						self.assertListEqual(UpperCAmelCase_       ,   UpperCAmelCase_							)

						SCREAMING_SNAKE_CASE__													= tokenizer.encode(UpperCAmelCase_       ,   add_special_tokens=UpperCAmelCase_							)
						SCREAMING_SNAKE_CASE__													= rust_tokenizer.encode(UpperCAmelCase_       ,   add_special_tokens=UpperCAmelCase_							)
						self.assertListEqual(UpperCAmelCase_       ,   UpperCAmelCase_							)

						SCREAMING_SNAKE_CASE__													= self.get_rust_tokenizer()
						SCREAMING_SNAKE_CASE__													= tokenizer.encode(UpperCAmelCase_							)
						SCREAMING_SNAKE_CASE__													= rust_tokenizer.encode(UpperCAmelCase_							)
						self.assertListEqual(UpperCAmelCase_       ,   UpperCAmelCase_							)
	def      A_				(			self							:   Union[str, Any]							):
						SCREAMING_SNAKE_CASE__													= 'This is a test'
						SCREAMING_SNAKE_CASE__													= [13, 1, 4398, 25, 21, 1289]
						SCREAMING_SNAKE_CASE__													= ['▁', 'T', 'his', '▁is', '▁a', '▁test']
						SCREAMING_SNAKE_CASE__													= ['▁', '<unk>', 'his', '▁is', '▁a', '▁test']

						SCREAMING_SNAKE_CASE__													= DebertaVaTokenizer(UpperCAmelCase_       ,   keep_accents=UpperCAmelCase_							)
						SCREAMING_SNAKE_CASE__													= DebertaVaTokenizerFast(UpperCAmelCase_       ,   keep_accents=UpperCAmelCase_							)

						SCREAMING_SNAKE_CASE__													= tokenizer.encode(UpperCAmelCase_       ,   add_special_tokens=UpperCAmelCase_							)
						self.assertListEqual(UpperCAmelCase_       ,   UpperCAmelCase_							)
						SCREAMING_SNAKE_CASE__													= tokenizer.tokenize(UpperCAmelCase_							)
						self.assertListEqual(UpperCAmelCase_       ,   UpperCAmelCase_							)
						SCREAMING_SNAKE_CASE__													= tokenizer.convert_ids_to_tokens(UpperCAmelCase_							)
						self.assertListEqual(UpperCAmelCase_       ,   UpperCAmelCase_							)

						SCREAMING_SNAKE_CASE__													= rust_tokenizer.encode(UpperCAmelCase_       ,   add_special_tokens=UpperCAmelCase_							)
						self.assertListEqual(UpperCAmelCase_       ,   UpperCAmelCase_							)
						SCREAMING_SNAKE_CASE__													= rust_tokenizer.tokenize(UpperCAmelCase_							)
						self.assertListEqual(UpperCAmelCase_       ,   UpperCAmelCase_							)
						SCREAMING_SNAKE_CASE__													= rust_tokenizer.convert_ids_to_tokens(UpperCAmelCase_							)
						self.assertListEqual(UpperCAmelCase_       ,   UpperCAmelCase_							)

						# fmt: off
						SCREAMING_SNAKE_CASE__													= 'I was born in 92000, and this is falsé.'
						SCREAMING_SNAKE_CASE__													= [13, 1, 23, 386, 19, 561, 3050, 15, 17, 48, 25, 8256, 18, 1, 9]
						SCREAMING_SNAKE_CASE__													= ['▁', 'I', '▁was', '▁born', '▁in', '▁9', '2000', ',', '▁and', '▁this', '▁is', '▁fal', 's', 'é', '.', ]
						SCREAMING_SNAKE_CASE__													= ['▁', '<unk>', '▁was', '▁born', '▁in', '▁9', '2000', ',', '▁and', '▁this', '▁is', '▁fal', 's', '<unk>', '.', ]
						# fmt: on

						SCREAMING_SNAKE_CASE__													= tokenizer.encode(UpperCAmelCase_       ,   add_special_tokens=UpperCAmelCase_							)
						self.assertListEqual(UpperCAmelCase_       ,   UpperCAmelCase_							)
						SCREAMING_SNAKE_CASE__													= tokenizer.tokenize(UpperCAmelCase_							)
						self.assertListEqual(UpperCAmelCase_       ,   UpperCAmelCase_							)
						SCREAMING_SNAKE_CASE__													= tokenizer.convert_ids_to_tokens(UpperCAmelCase_							)
						self.assertListEqual(UpperCAmelCase_       ,   UpperCAmelCase_							)

						SCREAMING_SNAKE_CASE__													= rust_tokenizer.encode(UpperCAmelCase_       ,   add_special_tokens=UpperCAmelCase_							)
						self.assertListEqual(UpperCAmelCase_       ,   UpperCAmelCase_							)
						SCREAMING_SNAKE_CASE__													= rust_tokenizer.tokenize(UpperCAmelCase_							)
						self.assertListEqual(UpperCAmelCase_       ,   UpperCAmelCase_							)
						SCREAMING_SNAKE_CASE__													= rust_tokenizer.convert_ids_to_tokens(UpperCAmelCase_							)
						self.assertListEqual(UpperCAmelCase_       ,   UpperCAmelCase_							)
	def      A_				(			self							:   Optional[Any]							):
						SCREAMING_SNAKE_CASE__													= DebertaVaTokenizer(UpperCAmelCase_							)

						SCREAMING_SNAKE_CASE__													= tokenizer.encode('sequence builders'							)
						SCREAMING_SNAKE_CASE__													= tokenizer.encode('multi-sequence build'							)

						SCREAMING_SNAKE_CASE__													= tokenizer.build_inputs_with_special_tokens(UpperCAmelCase_							)
						SCREAMING_SNAKE_CASE__													= tokenizer.build_inputs_with_special_tokens(UpperCAmelCase_       ,   UpperCAmelCase_							)

						self.assertEqual([tokenizer.cls_token_id] + text + [tokenizer.sep_token_id]       ,   UpperCAmelCase_							)
						self.assertEqual(
						    [tokenizer.cls_token_id] + text + [tokenizer.sep_token_id] + text_a + [tokenizer.sep_token_id]       ,   UpperCAmelCase_       ,   )





	@slow
	def      A_				(			self							:   Optional[Any]							):
						# fmt: off
						SCREAMING_SNAKE_CASE__													= {'input_ids': [[1, 39867, 36, 19390, 486, 27, 35052, 81436, 18, 60685, 1225, 7, 35052, 81436, 18, 9367, 16899, 18, 15937, 53, 594, 773, 18, 16287, 30465, 36, 15937, 6, 41139, 38, 36979, 60763, 191, 6, 34132, 99, 6, 50538, 390, 43230, 6, 34132, 2779, 20850, 14, 699, 1072, 1194, 36, 382, 10901, 53, 7, 699, 1072, 2084, 36, 20422, 630, 53, 19, 105, 3049, 1896, 1053, 16899, 1506, 11, 37978, 4243, 7, 1237, 31869, 200, 16566, 654, 6, 35052, 81436, 7, 55630, 13593, 4, 2], [1, 26, 15011, 13, 667, 8, 1053, 18, 23611, 1237, 72356, 12820, 34, 104134, 1209, 35, 13313, 6627, 21, 202, 347, 7, 164, 2399, 11, 46, 4485, 4, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0], [1, 5, 1232, 2864, 15785, 14951, 105, 5, 8581, 1250, 4, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], 'token_type_ids': [[0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0], [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0], [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], 'attention_mask': [[1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1], [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0], [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]]}  # noqa: E501
						# fmt: on

						self.tokenizer_integration_test_util(
						    expected_encoding=UpperCAmelCase_       ,   model_name='microsoft/deberta-v2-xlarge'       ,   revision='ad6e42c1532ddf3a15c39246b63f5559d558b670'       ,   )





