mirror of https://github.com/coqui-ai/TTS.git
Make style
This commit is contained in:
parent
37b708dac7
commit
f4c88ed677
|
@ -4,10 +4,10 @@ Much of this code is adapted from Andrej Karpathy's NanoGPT
|
||||||
"""
|
"""
|
||||||
import math
|
import math
|
||||||
from dataclasses import dataclass
|
from dataclasses import dataclass
|
||||||
from coqpit import Coqpit
|
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
import torch.nn as nn
|
import torch.nn as nn
|
||||||
|
from coqpit import Coqpit
|
||||||
from torch.nn import functional as F
|
from torch.nn import functional as F
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -251,7 +251,7 @@ class ModelManager(object):
|
||||||
else:
|
else:
|
||||||
self._download_zip_file(model_item["github_rls_url"], output_path, self.progress_bar)
|
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):
|
if isinstance(model_item["hf_url"], list):
|
||||||
self._download_model_files(model_item["hf_url"], output_path, self.progress_bar)
|
self._download_model_files(model_item["hf_url"], output_path, self.progress_bar)
|
||||||
else:
|
else:
|
||||||
|
|
Loading…
Reference in New Issue