Skip to content

Commit

Permalink
correction. model does load more than wav
Browse files Browse the repository at this point in the history
print to console when using cpu
  • Loading branch information
Dont-Copy-That-Floppy committed Apr 28, 2020
1 parent 485d285 commit b333e73
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions demo_cli.py
Original file line number Diff line number Diff line change
Expand Up @@ -41,7 +41,7 @@
## Print some environment information (for debugging purposes)
print("Running a test of your configuration...\n")
if args.cpu:
encoder.load_model(args.enc_model_fpath)
print("Using CPU for inference.")
elif torch.cuda.is_available():
device_id = torch.cuda.current_device()
gpu_properties = torch.cuda.get_device_properties(device_id)
Expand Down Expand Up @@ -120,8 +120,9 @@
num_generated = 0
while True:
try:
# Get the reference audio filepath
message = "Reference voice: enter an audio filepath of a voice to be cloned (.wav only):\n"
# Get the reference audio filepath
message = "Reference voice: enter an audio filepath of a voice to be cloned (mp3, " \
"wav, m4a, flac, ...):\n"
in_fpath = Path(input(message).replace("\"", "").replace("\'", ""))


Expand Down

0 comments on commit b333e73

Please sign in to comment.