This commit is contained in:
2023-02-18 21:01:47 +08:00
parent 9a218853ea
commit 607d958a9d

View File

@@ -230,28 +230,9 @@ def main_dev(opt):
print(data) # [{'model': '768-v-ema', 'prompt': '一只猫', 'watermark': '0'}, {'model': '768-v-ema', 'prompt': '一只狗', 'watermark': '0'}] print(data) # [{'model': '768-v-ema', 'prompt': '一只猫', 'watermark': '0'}, {'model': '768-v-ema', 'prompt': '一只狗', 'watermark': '0'}]
# 遍历 data 返回dict # 遍历 data 返回dict
for item in data: for item in data:
print(item) # {'model': '768-v-ema', 'prompt': '一只猫', 'watermark': '0'} print(item)
''' update_task_status(item, "running", 0) # 更新任务状态为运行中
task: {
model: 'SD2',
ckpt: 'latest',
prompt: '猫猫',
number: 1,
tid: '06ruxroiuo3u',
uid: 1234567890,
status: 'waiting',
createdAt: 1676660844133,
remove: '狗狗',
w: 512,
h: 512,
seed: 0,
sampler: 'pndm',
prompt_guidance: 0.5,
quality_details: 25,
image: '',
data: null
}
'''
# 设置参数 # 设置参数
if 'prompt' in item: opt.prompt = item['prompt'] # 描述 if 'prompt' in item: opt.prompt = item['prompt'] # 描述
if 'n_samples' in item: opt.n_samples = item['n_samples'] # 列数 if 'n_samples' in item: opt.n_samples = item['n_samples'] # 列数