This commit is contained in:
2023-02-18 21:50:55 +08:00
parent f92d9e9de6
commit 3ab2cb75ea

View File

@@ -362,7 +362,7 @@ def main_dev(opt):
precision_scope = autocast if opt.precision == "autocast" or opt.bf16 else nullcontext
with torch.no_grad(), precision_scope(opt.device), model.ema_scope():
#all_samples = list()
# 执行指定的任务组数 (row)(item['number'])
# 执行指定的任务批次 (row)(item['number'])
for n in trange(1, desc="Sampling"):
print("Sampling:", n)
for prompts in tqdm(data, desc="data"):