Spaces:
Running
on
Zero
Running
on
Zero
Update app_onnx.py
Browse files- 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=
|
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")
|