DEBUG
This commit is contained in:
@@ -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
|
||||
|
Reference in New Issue
Block a user