lint(api): sort schedulers, imports
This commit is contained in:
parent
6391f4b2af
commit
64c78b887e
22
api/serve.py
22
api/serve.py
|
@ -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)
|
||||||
|
|
Loading…
Reference in New Issue