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