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: