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

[Wav2Vec2] Fix torch srcipt (#24062)

Browse Source 
* [Wav2Vec2] Fix torch srcipt

* fix more
This commit is contained in:
Patrick von Platen
2023-06-07 13:27:07 +02:00
committed by GitHub
parent 612b2a1a6d
commit 52972e70c7
2 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/models/wav2vec2/test_modeling_wav2vec2.py
View File

@@ -297,7 +297,7 @@ class Wav2Vec2ModelTester:
config.adapter_attn_dim = 16
model = Wav2Vec2ForCTC(config=config)
self.parent.assertIsNotNone(model._adapters)
self.parent.assertIsNotNone(model._get_adapters())
model.to(torch_device)
model.eval()
@@ -1146,7 +1146,7 @@ class Wav2Vec2RobustModelTest(ModelTesterMixin, unittest.TestCase):
model = Wav2Vec2ForCTC.from_pretrained(tempdir)
logits = get_logits(model, input_features)
adapter_weights = model._adapters
adapter_weights = model._get_adapters()
# save safe weights
safe_filepath = os.path.join(tempdir, WAV2VEC2_ADAPTER_SAFE_FILE.format("eng"))
@@ -1168,7 +1168,7 @@ class Wav2Vec2RobustModelTest(ModelTesterMixin, unittest.TestCase):
model = Wav2Vec2ForCTC.from_pretrained(tempdir)
logits = get_logits(model, input_features)
adapter_weights = model._adapters
adapter_weights = model._get_adapters()
# save pt weights
pt_filepath = os.path.join(tempdir, WAV2VEC2_ADAPTER_PT_FILE.format("eng"))