gameui/drawing
2
0
Fork 0
Code Issues 1 Pull Requests Packages Projects Releases Wiki Activity

批次

Browse Source
This commit is contained in:
桜華
2023-02-18 21:50:55 +08:00
parent f92d9e9de6
commit 3ab2cb75ea
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
server.py
View File

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