apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

lint(api): sort schedulers, imports

This commit is contained in:
Sean Sube 2023-01-05 17:23:37 -06:00
parent 6391f4b2af
commit 64c78b887e
1 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
api/serve.py
View File

@ -1,12 +1,12 @@
from diffusers import OnnxStableDiffusionPipeline from diffusers import OnnxStableDiffusionPipeline
from diffusers import ( from diffusers import (
DDPMScheduler,
DDIMScheduler, DDIMScheduler,
PNDMScheduler, DDPMScheduler,
LMSDiscreteScheduler, DPMSolverMultistepScheduler,
EulerDiscreteScheduler, EulerDiscreteScheduler,
EulerAncestralDiscreteScheduler, EulerAncestralDiscreteScheduler,
DPMSolverMultistepScheduler, LMSDiscreteScheduler,
PNDMScheduler,
) )
from flask import Flask, make_response, request, send_file from flask import Flask, make_response, request, send_file
from stringcase import spinalcase from stringcase import spinalcase
@ -31,14 +31,14 @@ model_path = environ.get('ONNX_WEB_MODEL_PATH', "../models/stable-diffusion-onnx
output_path = environ.get('ONNX_WEB_OUTPUT_PATH', "../outputs") output_path = environ.get('ONNX_WEB_OUTPUT_PATH', "../outputs")
# schedulers # schedulers
scheduler_list = { pipeline_schedulers = {
'ddpm': DDPMScheduler.from_pretrained(model_path, subfolder="scheduler"),
'ddim': DDIMScheduler.from_pretrained(model_path, subfolder="scheduler"), 'ddim': DDIMScheduler.from_pretrained(model_path, subfolder="scheduler"),
'pndm': PNDMScheduler.from_pretrained(model_path, subfolder="scheduler"), 'ddpm': DDPMScheduler.from_pretrained(model_path, subfolder="scheduler"),
'lms-discrete': LMSDiscreteScheduler.from_pretrained(model_path, subfolder="scheduler"),
'euler-a': EulerAncestralDiscreteScheduler.from_pretrained(model_path, subfolder="scheduler"),
'euler': EulerDiscreteScheduler.from_pretrained(model_path, subfolder="scheduler"),
'dpm-multi': DPMSolverMultistepScheduler.from_pretrained(model_path, subfolder="scheduler"), 'dpm-multi': DPMSolverMultistepScheduler.from_pretrained(model_path, subfolder="scheduler"),
'euler': EulerDiscreteScheduler.from_pretrained(model_path, subfolder="scheduler"),
'euler-a': EulerAncestralDiscreteScheduler.from_pretrained(model_path, subfolder="scheduler"),
'lms-discrete': LMSDiscreteScheduler.from_pretrained(model_path, subfolder="scheduler"),
'pndm': PNDMScheduler.from_pretrained(model_path, subfolder="scheduler"),
} }
@ -86,7 +86,7 @@ def txt2img():
prompt = request.args.get('prompt', default_prompt) prompt = request.args.get('prompt', default_prompt)
scheduler = get_from_map(request.args, 'scheduler', scheduler = get_from_map(request.args, 'scheduler',
scheduler_list, 'euler-a') pipeline_schedulers, 'euler-a')
cfg = get_and_clamp(request.args, 'cfg', default_cfg, max_cfg, 0) cfg = get_and_clamp(request.args, 'cfg', default_cfg, max_cfg, 0)
steps = get_and_clamp(request.args, 'steps', default_steps, max_steps) steps = get_and_clamp(request.args, 'steps', default_steps, max_steps)
height = get_and_clamp(request.args, 'height', default_height, max_height) height = get_and_clamp(request.args, 'height', default_height, max_height)