This commit is contained in:
2023-02-18 22:03:07 +08:00
parent 3ab2cb75ea
commit 53cc9d52d3

View File

@@ -239,10 +239,10 @@ def main_dev(opt):
# 设置参数
if 'prompt' in item:
opt.prompt = item['prompt'] # 描述
if 'n_samples' in item:
if 'number' in item:
opt.n_samples = item['number'] # 列数
if 'n_rows' in item:
opt.n_rows = item['n_rows'] # 行数
#if 'n_rows' in item:
# opt.n_rows = item['n_rows'] # 行数
if 'scale' in item:
opt.scale = item['scale'] # 比例
@@ -276,7 +276,7 @@ def main_dev(opt):
wm_encoder.set_watermark('bytes', wm.encode('utf-8'))
# x
batch_size = opt.n_samples
n_rows = opt.n_rows if opt.n_rows > 0 else batch_size
#n_rows = opt.n_rows if opt.n_rows > 0 else batch_size
if not opt.from_file:
prompt = opt.prompt
assert prompt is not None