diff --git a/TTS/tts/layers/bark/model.py b/TTS/tts/layers/bark/model.py index 81117b3e..bcc87a4b 100644 --- a/TTS/tts/layers/bark/model.py +++ b/TTS/tts/layers/bark/model.py @@ -4,10 +4,10 @@ Much of this code is adapted from Andrej Karpathy's NanoGPT """ import math from dataclasses import dataclass -from coqpit import Coqpit import torch import torch.nn as nn +from coqpit import Coqpit from torch.nn import functional as F diff --git a/TTS/utils/manage.py b/TTS/utils/manage.py index dc0c7b68..f9968910 100644 --- a/TTS/utils/manage.py +++ b/TTS/utils/manage.py @@ -251,7 +251,7 @@ class ModelManager(object): else: self._download_zip_file(model_item["github_rls_url"], output_path, self.progress_bar) - def _download_hf_model(self, model_item:Dict, output_path: str): + def _download_hf_model(self, model_item: Dict, output_path: str): if isinstance(model_item["hf_url"], list): self._download_model_files(model_item["hf_url"], output_path, self.progress_bar) else: