[SpeechSynthesis/Tacotron2] Error while export ONNX models #487

root@32fdccde2ef1:/workspace/tacotron2# python exports/export_tacotron2_onnx.py --tacotron2 ./checkpoints/tacotron2_1032590_6000_amp -o output/ --fp16
Traceback (most recent call last):
  File "exports/export_tacotron2_onnx.py", line 405, in <module>
    main()
  File "exports/export_tacotron2_onnx.py", line 300, in main
    amp_run=args.fp16)
TypeError: load_and_setup_model() missing 1 required positional argument: 'cpu_run'

root@32fdccde2ef1:/workspace/tacotron2# python exports/export_waveglow_onnx.py --waveglow ./checkpoints/waveglow_1076430_14000_amp --wn-channels 256 -o output/ --fp16
Traceback (most recent call last):
  File "exports/export_waveglow_onnx.py", line 216, in <module>
    main()
  File "exports/export_waveglow_onnx.py", line 213, in main
    export_onnx(parser, args)
  File "exports/export_waveglow_onnx.py", line 168, in export_onnx
    args.fp16, forward_is_infer=False)
TypeError: load_and_setup_model() missing 1 required positional argument: 'cpu_run'