fix: eval when generate_with_interleaved

This commit is contained in:
lionHC
2024-06-12 10:01:01 +08:00
committed by GitHub
parent d9cf8604f3
commit 235132f39b
2 changed files with 2 additions and 1 deletions

View File

@@ -56,7 +56,7 @@ def parse_args():
parser.add_argument("--device", type=str, default="cuda:0")
parser.add_argument('--batchsize', type=int, default=1, help='Batch size for processing.')
parser.add_argument("--ckpt", type=str, default="")
parser.add_argument("--ckpt", type=str, default=None)
args = parser.parse_args()
return args