apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

add SDXL pipelines to list

This commit is contained in:
Sean Sube 2023-08-06 10:05:03 -05:00
parent c81dcf16e6
commit a201a74b40
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
api/onnx_web/diffusers/load.py
View File

@ -3,6 +3,7 @@ from os import path
from typing import Any, List, Optional, Tuple
from onnx import load_model
from optimum import ORTStableDiffusionXLPipeline, ORTStableDiffusionXLImg2ImgPipeline
from transformers import CLIPTokenizer
from ..constants import ONNX_MODEL
@ -47,10 +48,12 @@ logger = getLogger(__name__)
available_pipelines = {
"controlnet": OnnxStableDiffusionControlNetPipeline,
"img2img": OnnxStableDiffusionImg2ImgPipeline,
"img2img-sdxl": ORTStableDiffusionXLImg2ImgPipeline,
"inpaint": OnnxStableDiffusionInpaintPipeline,
"lpw": OnnxStableDiffusionLongPromptWeightingPipeline,
"panorama": OnnxStableDiffusionPanoramaPipeline,
"pix2pix": OnnxStableDiffusionInstructPix2PixPipeline,
"txt2img-sdxl": ORTStableDiffusionXLPipeline,
"txt2img": OnnxStableDiffusionPipeline,
"upscale": OnnxStableDiffusionUpscalePipeline,
}

4
api/requirements/base.txt
View File

@ -7,10 +7,12 @@ accelerate==0.16.0
coloredlogs==15.0.1
controlnet_aux==0.0.2
diffusers==0.15.1
imwatermark
mediapipe==0.9.2.1
omegaconf==2.3.0
onnx==1.13.0
# onnxruntime has many platform-specific packages
optimum
safetensors==0.2.8
timm==0.6.13
transformers==4.26.1
@ -31,4 +33,4 @@ jsonschema==4.17.3
piexif==1.1.3
pyyaml==6.0
setproctitle==1.3.2
waitress==2.1.2
waitress==2.1.2