mirror of
https://github.com/pstrueb/piper.git
synced 2026-04-18 06:15:30 +00:00
Bump version
This commit is contained in:
@@ -13,7 +13,7 @@ _LOGGER = logging.getLogger("piper_train.export_onnx")
|
||||
OPSET_VERSION = 15
|
||||
|
||||
|
||||
def main():
|
||||
def main() -> None:
|
||||
"""Main entry point"""
|
||||
torch.manual_seed(1234)
|
||||
|
||||
@@ -76,7 +76,7 @@ def main():
|
||||
)
|
||||
sequence_lengths = torch.LongTensor([sequences.size(1)])
|
||||
|
||||
sid: Optional[int] = None
|
||||
sid: Optional[torch.LongTensor] = None
|
||||
if num_speakers > 1:
|
||||
sid = torch.LongTensor([0])
|
||||
|
||||
@@ -89,7 +89,7 @@ def main():
|
||||
model=model_g,
|
||||
args=dummy_input,
|
||||
f=str(args.output),
|
||||
verbose=True,
|
||||
verbose=False,
|
||||
opset_version=OPSET_VERSION,
|
||||
input_names=["input", "input_lengths", "scales", "sid"],
|
||||
output_names=["output"],
|
||||
|
||||
Reference in New Issue
Block a user