mirror of https://github.com/coqui-ai/TTS.git
Merge pull request #527 from WeberJulian/dev
fix: fixing the RenamingUnpickler fix
This commit is contained in:
commit
82376298d9
|
@ -6,9 +6,7 @@ import pickle as pickle_tts
|
||||||
class RenamingUnpickler(pickle_tts.Unpickler):
|
class RenamingUnpickler(pickle_tts.Unpickler):
|
||||||
"""Overload default pickler to solve module renaming problem"""
|
"""Overload default pickler to solve module renaming problem"""
|
||||||
def find_class(self, module, name):
|
def find_class(self, module, name):
|
||||||
if 'TTS' in module:
|
return super().find_class(module.replace('mozilla_voice_tts', 'TTS'), name)
|
||||||
module = module.replace('TTS', 'TTS')
|
|
||||||
return super().find_class(module, name)
|
|
||||||
|
|
||||||
|
|
||||||
class AttrDict(dict):
|
class AttrDict(dict):
|
||||||
|
|
Loading…
Reference in New Issue