coqui-ai
/
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

pull/2870/head
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 torch
from torch import nn, optim
from torch.serialization import MAP_LOCATION
class HubertTokenizer(nn.Module):
@ -103,7 +102,7 @@ class HubertTokenizer(nn.Module):
model_zip.close()
@staticmethod
def load_from_checkpoint(path, map_location: MAP_LOCATION = None):
def load_from_checkpoint(path, map_location = None):
old = True
with ZipFile(path) as model_zip:
filesMatch = [file for file in model_zip.namelist() if file.endswith("/.info")]