SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[EnCodec] Changes for 32kHz ckpt (#24296)

Browse Source 
* [EnCodec] Changes for 32kHz ckpt

* Update src/transformers/models/encodec/convert_encodec_checkpoint_to_pytorch.py

* Update src/transformers/models/encodec/convert_encodec_checkpoint_to_pytorch.py
This commit is contained in:
Sanchit Gandhi
2023-06-15 14:36:19 +01:00
committed by GitHub
parent 01b55779d3
commit 4124a09f8b
4 changed files with 29 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/models/encodec/test_modeling_encodec.py
View File

@@ -385,6 +385,11 @@ class EncodecModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase)
msg=f"Parameter {name} of model {model_class} seems not properly initialized",
)
def test_identity_shortcut(self):
config, inputs_dict = self.model_tester.prepare_config_and_inputs()
config.use_conv_shortcut = False
self.model_tester.create_and_check_model_forward(config, inputs_dict)
def normalize(arr):
norm = np.linalg.norm(arr)