Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix import error with Bark

This commit is contained in:
Eren G??lge 2023-08-11 03:33:59 +02:00
parent c87377b713
commit 9a8352b8da
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
TTS/tts/layers/bark/hubert/tokenizer.py
View File

@ -11,7 +11,6 @@ from zipfile import ZipFile
import numpy import numpy
import torch import torch
from torch import nn, optim from torch import nn, optim
from torch.serialization import MAP_LOCATION
class HubertTokenizer(nn.Module): class HubertTokenizer(nn.Module):
@ -103,7 +102,7 @@ class HubertTokenizer(nn.Module):
model_zip.close() model_zip.close()
@staticmethod @staticmethod
def load_from_checkpoint(path, map_location: MAP_LOCATION = None): def load_from_checkpoint(path, map_location = None):
old = True old = True
with ZipFile(path) as model_zip: with ZipFile(path) as model_zip:
filesMatch = [file for file in model_zip.namelist() if file.endswith("/.info")] filesMatch = [file for file in model_zip.namelist() if file.endswith("/.info")]