
'''simple docstring'''





import unittest

from transformers import SPIECE_UNDERLINE, XLNetTokenizer, XLNetTokenizerFast
from transformers.testing_utils import get_tests_dir, require_sentencepiece, require_tokenizers, slow

from ...test_tokenization_common import TokenizerTesterMixin


UpperCAmelCase    					=   get_tests_dir("fixtures/test_sentencepiece.model")

@require_sentencepiece
@require_tokenizers
class 						a   (				__magic_name__				,unittest.TestCase    ):
				_snake_case		    = XLNetTokenizer
				_snake_case		    = XLNetTokenizerFast
				_snake_case		    = True
				_snake_case		    = True




				def 	__snake_case							(							self	:				Dict      ):
										super().setUp()

										# We have a SentencePiece fixture for testing
										snake_case       :     Optional[Any] =    XLNetTokenizer(SCREAMING_SNAKE_CASE_,    keep_accents=SCREAMING_SNAKE_CASE_      )
										tokenizer.sanitize_special_tokens()
										tokenizer.save_pretrained(self.tmpdirname      )




				def 	__snake_case							(							self	:				str      ):
										snake_case       :     Optional[Any] =    '''<s>'''
										snake_case       :     str =    1

										self.assertEqual(self.get_tokenizer()._convert_token_to_id(SCREAMING_SNAKE_CASE_      ),    SCREAMING_SNAKE_CASE_      )
										self.assertEqual(self.get_tokenizer()._convert_id_to_token(SCREAMING_SNAKE_CASE_      ),    SCREAMING_SNAKE_CASE_      )




				def 	__snake_case							(							self	:				Any      ):
										snake_case       :     Any =    list(self.get_tokenizer().get_vocab().keys()      )

										self.assertEqual(vocab_keys[0],    '''<unk>'''      )
										self.assertEqual(vocab_keys[1],    '''<s>'''      )
										self.assertEqual(vocab_keys[-1],    '''<eod>'''      )
										self.assertEqual(len(SCREAMING_SNAKE_CASE_      ),    10_06      )




				def 	__snake_case							(							self	:				Tuple      ):
										self.assertEqual(self.get_tokenizer().vocab_size,    10_00      )




				def 	__snake_case							(							self	:				str      ):
										snake_case       :     Union[str, Any] =    XLNetTokenizer(SCREAMING_SNAKE_CASE_,    keep_accents=SCREAMING_SNAKE_CASE_      )

										snake_case       :     Optional[Any] =    tokenizer.tokenize('''This is a test'''      )
										self.assertListEqual(SCREAMING_SNAKE_CASE_,    ['''▁This''', '''▁is''', '''▁a''', '''▁t''', '''est''']      )

										self.assertListEqual(tokenizer.convert_tokens_to_ids(SCREAMING_SNAKE_CASE_      ),    [2_85, 46, 10, 1_70, 3_82]      )

										snake_case       :     Optional[int] =    tokenizer.tokenize('''I was born in 92000, and this is falsé.'''      )
										self.assertListEqual(
										    SCREAMING_SNAKE_CASE_,    [
										        SPIECE_UNDERLINE + '''I''',
										        SPIECE_UNDERLINE + '''was''',
										        SPIECE_UNDERLINE + '''b''',
										        '''or''',
										        '''n''',
										        SPIECE_UNDERLINE + '''in''',
										        SPIECE_UNDERLINE + '''''',
										        '''9''',
										        '''2''',
										        '''0''',
										        '''0''',
										        '''0''',
										        ''',''',
										        SPIECE_UNDERLINE + '''and''',
										        SPIECE_UNDERLINE + '''this''',
										        SPIECE_UNDERLINE + '''is''',
										        SPIECE_UNDERLINE + '''f''',
										        '''al''',
										        '''s''',
										        '''é''',
										        '''.''',
										    ],    )
										snake_case       :     Tuple =    tokenizer.convert_tokens_to_ids(SCREAMING_SNAKE_CASE_      )
										self.assertListEqual(SCREAMING_SNAKE_CASE_,    [8, 21, 84, 55, 24, 19, 7, 0, 6_02, 3_47, 3_47, 3_47, 3, 12, 66, 46, 72, 80, 6, 0, 4]      )

										snake_case       :     Any =    tokenizer.convert_ids_to_tokens(SCREAMING_SNAKE_CASE_      )
										self.assertListEqual(
										    SCREAMING_SNAKE_CASE_,    [
										        SPIECE_UNDERLINE + '''I''',
										        SPIECE_UNDERLINE + '''was''',
										        SPIECE_UNDERLINE + '''b''',
										        '''or''',
										        '''n''',
										        SPIECE_UNDERLINE + '''in''',
										        SPIECE_UNDERLINE + '''''',
										        '''<unk>''',
										        '''2''',
										        '''0''',
										        '''0''',
										        '''0''',
										        ''',''',
										        SPIECE_UNDERLINE + '''and''',
										        SPIECE_UNDERLINE + '''this''',
										        SPIECE_UNDERLINE + '''is''',
										        SPIECE_UNDERLINE + '''f''',
										        '''al''',
										        '''s''',
										        '''<unk>''',
										        '''.''',
										    ],    )




				def 	__snake_case							(							self	:				List[Any]      ):
										snake_case       :     Any =    XLNetTokenizer(SCREAMING_SNAKE_CASE_,    do_lower_case=SCREAMING_SNAKE_CASE_      )
										snake_case       :     Optional[Any] =    tokenizer.tokenize('''I was born in 92000, and this is falsé.'''      )
										self.assertListEqual(
										    SCREAMING_SNAKE_CASE_,    [
										        SPIECE_UNDERLINE + '''''',
										        '''i''',
										        SPIECE_UNDERLINE + '''was''',
										        SPIECE_UNDERLINE + '''b''',
										        '''or''',
										        '''n''',
										        SPIECE_UNDERLINE + '''in''',
										        SPIECE_UNDERLINE + '''''',
										        '''9''',
										        '''2''',
										        '''0''',
										        '''0''',
										        '''0''',
										        ''',''',
										        SPIECE_UNDERLINE + '''and''',
										        SPIECE_UNDERLINE + '''this''',
										        SPIECE_UNDERLINE + '''is''',
										        SPIECE_UNDERLINE + '''f''',
										        '''al''',
										        '''se''',
										        '''.''',
										    ],    )
										self.assertListEqual(tokenizer.tokenize('''H\u00E9llo'''      ),    ['''▁he''', '''ll''', '''o''']      )




				def 	__snake_case							(							self	:				List[Any]      ):
										snake_case       :     Dict =    XLNetTokenizer(SCREAMING_SNAKE_CASE_,    do_lower_case=SCREAMING_SNAKE_CASE_      )
										snake_case       :     str =    tokenizer.tokenize('''I was born in 92000, and this is falsé.'''      )
										self.assertListEqual(
										    SCREAMING_SNAKE_CASE_,    [
										        SPIECE_UNDERLINE + '''I''',
										        SPIECE_UNDERLINE + '''was''',
										        SPIECE_UNDERLINE + '''b''',
										        '''or''',
										        '''n''',
										        SPIECE_UNDERLINE + '''in''',
										        SPIECE_UNDERLINE + '''''',
										        '''9''',
										        '''2''',
										        '''0''',
										        '''0''',
										        '''0''',
										        ''',''',
										        SPIECE_UNDERLINE + '''and''',
										        SPIECE_UNDERLINE + '''this''',
										        SPIECE_UNDERLINE + '''is''',
										        SPIECE_UNDERLINE + '''f''',
										        '''al''',
										        '''se''',
										        '''.''',
										    ],    )




				@slow
				def 	__snake_case							(							self	:				int      ):
										snake_case       :     str =    XLNetTokenizer.from_pretrained('''xlnet-base-cased'''      )

										snake_case       :     int =    tokenizer.encode('''sequence builders''',    add_special_tokens=SCREAMING_SNAKE_CASE_      )
										snake_case       :     List[Any] =    tokenizer.encode('''multi-sequence build''',    add_special_tokens=SCREAMING_SNAKE_CASE_      )

										snake_case       :     Union[str, Any] =    tokenizer.build_inputs_with_special_tokens(SCREAMING_SNAKE_CASE_      )
										snake_case       :     int =    tokenizer.build_inputs_with_special_tokens(SCREAMING_SNAKE_CASE_,    SCREAMING_SNAKE_CASE_      )

										assert encoded_sentence == text + [4, 3]
										assert encoded_pair == text + [4] + text_a + [4, 3]




				@slow
				def 	__snake_case							(							self	:				Tuple      ):
										# fmt: off
										snake_case       :     Optional[Any] =    {'''input_ids''': [[17, 2_14_42, 2_70, 17, 10, 1_46_45, 3_18, 34, 17, 45_46, 31_45, 7_87, 13, 77_52, 2_20_18, 23, 21, 17, 45_46, 31_45, 7_87, 13, 33_52, 1_44_31, 13, 55_00, 11, 11_76, 5_80, 13, 1_68_19, 47_97, 23, 17, 10, 1_71_35, 6_58, 19, 4_57, 79_32, 13, 1_84, 19, 31_54, 1_71_35, 64_68, 19, 14_04, 1_22_69, 19, 42_29, 53_56, 1_62_64, 46, 19, 17, 2_05_45, 1_03_95, 9, 9, 9, 11, 28, 64_21, 95_31, 2_07_29, 17, 10, 3_53, 1_70_22, 11, 21, 64_21, 95_31, 1_69_49, 17, 10, 1_15_09, 7_53, 11, 33, 95, 24_21, 73_85, 9_56, 1_44_31, 26_26, 25, 8_42, 73_85, 48_36, 21, 14_29, 22_72, 98_55, 31_20, 1_61, 2_47_38, 19, 1_32_03, 6_58, 2_18, 7_87, 21, 4_30, 1_84_82, 8_47, 26_37, 9, 4, 3], [5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 3_22, 2_21_78, 27, 10_64, 22, 9_56, 13, 1_11_01, 14_29, 58_54, 2_43_13, 1_89_53, 40, 4_22, 2_43_66, 68, 17_58, 37, 1_04_83, 1_42_57, 31, 2_07, 2_63, 21, 2_03, 37_73, 25, 71, 97_35, 9, 4, 3], [5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 32, 20_49, 34_42, 17, 1_38_94, 33_80, 23, 95, 18, 1_76_34, 22_88, 9, 4, 3]], '''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, 2], [3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 2], [3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 2]], '''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], [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 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, 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, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 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, 1, 1]]}  # noqa: E501
										# fmt: on

										self.tokenizer_integration_test_util(
										    expected_encoding=SCREAMING_SNAKE_CASE_,    model_name='''xlnet-base-cased''',    revision='''c841166438c31ec7ca9a106dee7bb312b73ae511''',    )





'''simple docstring'''





import argparse

import torch

# Step 1. clone https://github.com/microsoft/unilm
# Step 2. git checkout to https://github.com/microsoft/unilm/commit/b94ec76c36f02fb2b0bf0dcb0b8554a2185173cd
# Step 3. cd unilm
# Step 4. ln -s $(realpath wavlm/modules.py) ./  # create simlink
# import classes
from unilm.wavlm.WavLM import WavLM as WavLMOrig
from unilm.wavlm.WavLM import WavLMConfig as WavLMConfigOrig

from transformers import WavLMConfig, WavLMModel, logging


logging.set_verbosity_info()
UpperCAmelCase    					=   logging.get_logger(__name__)

UpperCAmelCase    					=   {
    "post_extract_proj": "feature_projection.projection",
    "encoder.pos_conv.0": "encoder.pos_conv_embed.conv",
    "self_attn.k_proj": "encoder.layers.*.attention.k_proj",
    "self_attn.v_proj": "encoder.layers.*.attention.v_proj",
    "self_attn.q_proj": "encoder.layers.*.attention.q_proj",
    "self_attn.out_proj": "encoder.layers.*.attention.out_proj",
    "self_attn.grep_linear": "encoder.layers.*.attention.gru_rel_pos_linear",
    "self_attn.relative_attention_bias": "encoder.layers.*.attention.rel_attn_embed",
    "self_attn.grep_a": "encoder.layers.*.attention.gru_rel_pos_const",
    "self_attn_layer_norm": "encoder.layers.*.layer_norm",
    "fc1": "encoder.layers.*.feed_forward.intermediate_dense",
    "fc2": "encoder.layers.*.feed_forward.output_dense",
    "final_layer_norm": "encoder.layers.*.final_layer_norm",
    "encoder.layer_norm": "encoder.layer_norm",
    "w2v_model.layer_norm": "feature_projection.layer_norm",
    "quantizer.weight_proj": "quantizer.weight_proj",
    "quantizer.vars": "quantizer.codevectors",
    "project_q": "project_q",
    "final_proj": "project_hid",
    "w2v_encoder.proj": "ctc_proj",
    "mask_emb": "masked_spec_embed",
}
UpperCAmelCase    					=   [
    "ctc_proj",
    "quantizer.weight_proj",
    "quantizer.codevectors",
    "project_q",
    "project_hid",
]

def      A (       A_							:					Dict    ,							A_							:					Any    ,							A_							:					List[str]    ,							A_							:					Tuple    ,							A_							:					Optional[Any]       ):
						for attribute in key.split('''.'''       ):
												snake_case       :     Tuple =    getattr(A_    ,							A_       )

						if weight_type is not None:
												snake_case       :     Optional[Any] =    getattr(A_    ,							A_       ).shape
						else:
												snake_case       :     Any =    hf_pointer.shape

						assert hf_shape == value.shape, (
						    F"""Shape of hf {key + '.' + weight_type if weight_type is not None else ''} is {hf_shape}, but should be"""
						    F""" {value.shape} for {full_name}"""
						)

						if weight_type == "weight":
												snake_case       :     Optional[Any] =    value
						elif weight_type == "weight_g":
												snake_case       :     Any =    value
						elif weight_type == "weight_v":
												snake_case       :     int =    value
						elif weight_type == "bias":
												snake_case       :     Any =    value
						else:
												snake_case       :     Optional[int] =    value

						logger.info(F"""{key + '.' + weight_type if weight_type is not None else ''} was initialized from {full_name}."""       )

def      A (       A_							:					str    ,							A_							:					str       ):
						snake_case       :     Dict =    []
						snake_case       :     Optional[Any] =    fairseq_model.state_dict()

						snake_case       :     int =    hf_model.feature_extractor

						for name, value in fairseq_dict.items():
												snake_case       :     List[Any] =    False
												if "conv_layers" in name:
																		load_conv_layer(
																		    A_    ,							A_    ,							A_    ,							A_    ,							hf_model.config.feat_extract_norm == '''group'''    ,							)
																		snake_case       :     int =    True
												else:
																		for key, mapped_key in MAPPING.items():
																								if key in name or key.split('''w2v_model.'''       )[-1] == name.split('''.'''       )[0]:
																														snake_case       :     Optional[int] =    True
																														if "*" in mapped_key:
																																				snake_case       :     Dict =    name.split(A_       )[0].split('''.'''       )[-2]
																																				snake_case       :     List[str] =    mapped_key.replace('''*'''    ,							A_       )
																														if "weight_g" in name:
																																				snake_case       :     Optional[Any] =    '''weight_g'''
																														elif "weight_v" in name:
																																				snake_case       :     Optional[Any] =    '''weight_v'''
																														elif "bias" in name and "relative_attention_bias" not in name:
																																				snake_case       :     Any =    '''bias'''
																														elif "weight" in name:
																																				# TODO: don't match quantizer.weight_proj
																																				snake_case       :     Optional[int] =    '''weight'''
																														else:
																																				snake_case       :     Union[str, Any] =    None

																														set_recursively(A_    ,							A_    ,							A_    ,							A_    ,							A_       )
																								continue
												if not is_used:
																		unused_weights.append(A_       )

						logger.warning(F"""Unused weights: {unused_weights}"""       )

def      A (       A_							:					Tuple    ,							A_							:					Any    ,							A_							:					int    ,							A_							:					Dict    ,							A_							:					Optional[Any]       ):
						snake_case       :     List[str] =    full_name.split('''conv_layers.'''       )[-1]
						snake_case       :     List[Any] =    name.split('''.'''       )
						snake_case       :     Dict =    int(items[0]       )
						snake_case       :     Any =    int(items[1]       )

						if type_id == 0:
												if "bias" in name:
																		assert value.shape == feature_extractor.conv_layers[layer_id].conv.bias.data.shape, (
																		    F"""{full_name} has size {value.shape}, but"""
																		    F""" {feature_extractor.conv_layers[layer_id].conv.bias.data.shape} was found."""
																		)
																		snake_case       :     Tuple =    value
																		logger.info(F"""Feat extract conv layer {layer_id} was initialized from {full_name}."""       )
												elif "weight" in name:
																		assert value.shape == feature_extractor.conv_layers[layer_id].conv.weight.data.shape, (
																		    F"""{full_name} has size {value.shape}, but"""
																		    F""" {feature_extractor.conv_layers[layer_id].conv.weight.data.shape} was found."""
																		)
																		snake_case       :     Dict =    value
																		logger.info(F"""Feat extract conv layer {layer_id} was initialized from {full_name}."""       )
						elif (type_id == 2 and not use_group_norm) or (type_id == 2 and layer_id == 0 and use_group_norm):
												if "bias" in name:
																		assert value.shape == feature_extractor.conv_layers[layer_id].layer_norm.bias.data.shape, (
																		    F"""{full_name} has size {value.shape}, but {feature_extractor[layer_id].layer_norm.bias.data.shape} was"""
																		    " found."
																		)
																		snake_case       :     str =    value
																		logger.info(F"""Feat extract layer norm weight of layer {layer_id} was initialized from {full_name}."""       )
												elif "weight" in name:
																		assert value.shape == feature_extractor.conv_layers[layer_id].layer_norm.weight.data.shape, (
																		    F"""{full_name} has size {value.shape}, but"""
																		    F""" {feature_extractor[layer_id].layer_norm.weight.data.shape} was found."""
																		)
																		snake_case       :     Tuple =    value
																		logger.info(F"""Feat extract layer norm weight of layer {layer_id} was initialized from {full_name}."""       )
						else:
												unused_weights.append(A_       )

@torch.no_grad()
def      A (       A_							:					Tuple    ,							A_							:					Any    ,							A_							:					Optional[Any]=None       ):
						# load the pre-trained checkpoints
						snake_case       :     Any =    torch.load(A_       )
						snake_case       :     List[str] =    WavLMConfigOrig(checkpoint['''cfg''']       )
						snake_case       :     Union[str, Any] =    WavLMOrig(A_       )
						model.load_state_dict(checkpoint['''model''']       )
						model.eval()

						if config_path is not None:
												snake_case       :     Union[str, Any] =    WavLMConfig.from_pretrained(A_       )
						else:
												snake_case       :     Any =    WavLMConfig()

						snake_case       :     Optional[int] =    WavLMModel(A_       )

						recursively_load_weights(A_    ,							A_       )

						hf_wavlm.save_pretrained(A_       )


if __name__ == "__main__":
						UpperCAmelCase    					=   argparse.ArgumentParser()
						parser.add_argument("--pytorch_dump_folder_path", default=None, type=str, help="Path to the output PyTorch model.")
						parser.add_argument("--checkpoint_path", default=None, type=str, help="Path to fairseq checkpoint")
						parser.add_argument("--config_path", default=None, type=str, help="Path to hf config.json of model to convert")
						UpperCAmelCase    					=   parser.parse_args()
						convert_wavlm_checkpoint(args.checkpoint_path, args.pytorch_dump_folder_path, args.config_path)

