Update the version from 1.0.0 to 1.0.1

Takaaki-Saeki · Feb 15, 2024 · cb2c382 · cb2c382
1 parent 4baff9c
commit cb2c382
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 1 deletion.
diff --git a/discrete_speech_metrics/speechbertscore.py b/discrete_speech_metrics/speechbertscore.py
@@ -1,10 +1,16 @@
 # Copyright 2024 Takaaki Saeki
 # MIT LICENSE (https://opensource.org/license/mit/)
 
+import logging
 import torchaudio
 import torch
 from transformers import HubertModel, Wav2Vec2Model, WavLMModel
 
+# In PyTorch 2+, a warning for checkpoint mismatch is raised.
+# But it should be a false alarm according to the following issue.
+# https://github.com/huggingface/transformers/issues/26796
+# I have added the following line to suppress the warning.
+logging.getLogger("transformers").setLevel(logging.ERROR)
 
 def bert_score(v_generated, v_reference):
     """

diff --git a/discrete_speech_metrics/speechbleu.py b/discrete_speech_metrics/speechbleu.py
@@ -1,6 +1,7 @@
 # Copyright 2024 Takaaki Saeki
 # MIT LICENSE (https://opensource.org/license/mit/)
 
+import logging
 from transformers import HubertModel
 import os
 import pathlib
@@ -13,6 +14,11 @@
 from nltk.translate.bleu_score import sentence_bleu
 nltk.download('punkt')
 
+# In PyTorch 2+, a warning for checkpoint mismatch is raised.
+# But it should be a false alarm according to the following issue.
+# https://github.com/huggingface/transformers/issues/26796
+# I have added the following line to suppress the warning.
+logging.getLogger("transformers").setLevel(logging.ERROR)
 
 def int_array_to_chinese_unicode(arr):
     """

diff --git a/discrete_speech_metrics/speechtokendistance.py b/discrete_speech_metrics/speechtokendistance.py
@@ -1,6 +1,7 @@
 # Copyright 2024 Takaaki Saeki
 # MIT LICENSE (https://opensource.org/license/mit/)
 
+import logging
 from transformers import HubertModel
 import os
 import pathlib
@@ -12,6 +13,11 @@
 from Levenshtein import distance as levenshtein_distance
 import jellyfish
 
+# In PyTorch 2+, a warning for checkpoint mismatch is raised.
+# But it should be a false alarm according to the following issue.
+# https://github.com/huggingface/transformers/issues/26796
+# I have added the following line to suppress the warning.
+logging.getLogger("transformers").setLevel(logging.ERROR)
 
 def int_array_to_chinese_unicode(arr):
     """

diff --git a/setup.py b/setup.py
@@ -2,7 +2,7 @@
 
 setup(
     name='discrete-speech-metrics',
-    version='1.0.0',
+    version='1.0.1',
     packages=find_packages(),
     install_requires=[
         'numpy>=1.20.3',