diff --git a/style_bert_vits2/models/utils/__init__.py b/style_bert_vits2/models/utils/__init__.py index 189f05e..b91c74a 100644 --- a/style_bert_vits2/models/utils/__init__.py +++ b/style_bert_vits2/models/utils/__init__.py @@ -11,10 +11,8 @@ import torch from numpy.typing import NDArray from style_bert_vits2.logging import logger -from style_bert_vits2.models.utils import ( - checkpoints, # type: ignore - safetensors, # type: ignore -) +from style_bert_vits2.models.utils import checkpoints # type: ignore +from style_bert_vits2.models.utils import safetensors # type: ignore if TYPE_CHECKING: diff --git a/style_bert_vits2/nlp/bert_models.py b/style_bert_vits2/nlp/bert_models.py index 926b372..639025b 100644 --- a/style_bert_vits2/nlp/bert_models.py +++ b/style_bert_vits2/nlp/bert_models.py @@ -251,7 +251,6 @@ def unload_tokenizer(language: Languages) -> None: language (Languages): アンロードする BERT トークナイザーの言語 """ - if language in __loaded_tokenizers: del __loaded_tokenizers[language] gc.collect()