breadlicker45 commited on
Commit
d8f2437
·
verified ·
1 Parent(s): 3a494b9

Update app_onnx.py

Browse files
Files changed (1) hide show
  1. app_onnx.py +1 -1
app_onnx.py CHANGED
@@ -453,7 +453,7 @@ if __name__ == "__main__":
453
  parser.add_argument("--port", type=int, default=7860, help="gradio server port")
454
  parser.add_argument("--device", type=str, default="cuda", help="device to run model")
455
  parser.add_argument("--batch", type=int, default=8, help="batch size")
456
- parser.add_argument("--max-gen", type=int, default=1024, help="max")
457
  opt = parser.parse_args()
458
  OUTPUT_BATCH_SIZE = opt.batch
459
  soundfont_path = hf_hub_download_retry(repo_id="skytnt/midi-model", filename="soundfont.sf2")
 
453
  parser.add_argument("--port", type=int, default=7860, help="gradio server port")
454
  parser.add_argument("--device", type=str, default="cuda", help="device to run model")
455
  parser.add_argument("--batch", type=int, default=8, help="batch size")
456
+ parser.add_argument("--max-gen", type=int, default=2048, help="max")
457
  opt = parser.parse_args()
458
  OUTPUT_BATCH_SIZE = opt.batch
459
  soundfont_path = hf_hub_download_retry(repo_id="skytnt/midi-model", filename="soundfont.sf2")