From bab2a25c96cfd0c80509240d559e6811712174d1 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=A1=9C=E8=8F=AF?= Date: Sat, 18 Feb 2023 22:10:43 +0800 Subject: [PATCH] DEBUG --- server.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/server.py b/server.py index c66b4f4..5f6eb14 100644 --- a/server.py +++ b/server.py @@ -241,6 +241,7 @@ def main_dev(opt): opt.prompt = item['prompt'] # 描述 if 'number' in item: opt.n_samples = item['number'] # 列数 + print(f"n_samples: {opt.n_samples}") #if 'n_rows' in item: # opt.n_rows = item['n_rows'] # 行数 if 'scale' in item: @@ -364,7 +365,7 @@ def main_dev(opt): #all_samples = list() # 执行指定的任务批次 (row)(item['number']) for n in trange(1, desc="Sampling"): - print("Sampling:", n) + print("Sampling:", data) for prompts in tqdm(data, desc="data"): uc = None if opt.scale != 1.0: