diff --git a/server.py b/server.py index a7eb903..6dd975d 100644 --- a/server.py +++ b/server.py @@ -230,28 +230,9 @@ def main_dev(opt): print(data) # [{'model': '768-v-ema', 'prompt': '一只猫', 'watermark': '0'}, {'model': '768-v-ema', 'prompt': '一只狗', 'watermark': '0'}] # 遍历 data 返回dict for item in data: - print(item) # {'model': '768-v-ema', 'prompt': '一只猫', 'watermark': '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 - } - ''' + print(item) + update_task_status(item, "running", 0) # 更新任务状态为运行中 + # 设置参数 if 'prompt' in item: opt.prompt = item['prompt'] # 描述 if 'n_samples' in item: opt.n_samples = item['n_samples'] # 列数