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